jmxpearson

jmxpearson's Stars

• taichi-dev/taichi

  Productive, portable, and performant GPU programming in Python.

  Language:C++26.4k2.3k

• google/evojax

  Language:Jupyter Notebook85791

• instadeepai/jumanji

  🕹️ A diverse suite of scalable reinforcement learning environments in JAX

  Language:Python66485

• RobertTLange/gymnax

  RL Environments in JAX 🌍

  Language:Python67562

• google/jaxonnxruntime

  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.

  Language:Python10216

• openrlbenchmark/openrlbenchmark

  Language:Python2079

• neurogym/neurogym

  A curated collection of neuroscience tasks with a common interface.

  Language:Python26250

• mikailkhona/Mod_Cog

  Language:Python11

• araffin/sbx

  SBX: Stable Baselines Jax (SB3 + Jax)

  Language:Python36335

• google-deepmind/mujoco

  Multi-Joint dynamics with Contact. A general purpose physics simulator.

  Language:Jupyter Notebook8.4k852

• google/brax

  Massively parallel rigidbody physics simulation on accelerator hardware.

  Language:Jupyter Notebook2.4k259

• vwxyzjn/cleanrl

  High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, TD3, SAC, PPG)

  Language:Python6k675

• state-spaces/mamba

  Mamba SSM architecture

  Language:Python13.6k1.2k

• PufferAI/PufferLib

  Simplifying reinforcement learning for complex game environments

  Language:C1.4k69

• wilson-labs/cola

  Compositional Linear Algebra

  Language:Python45230

• flatironinstitute/nemos

  NEural MOdelS, a statistical modeling framework for neuroscience.

  Language:Python838

• giant-axon/lu.i-neuron-pcb

  Language:AMPL486

• brandbci/brand

  Backend for Real-time Asynchronous Neural Decoding (BRAND)

  Language:Python265

• project-improv/improv

  Adaptive Platform for Neuroscience Experiments

  Language:Python3913

• vocalpy/vocalpy

  A core package for acoustic communication research in Python

  Language:Python337

• patrick-kidger/diffrax

  Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

  Language:Python1.5k136

• whitead/skunk

  Insert SVG images into matplotlib elements. Can be used to also compose matplotlib plots by nesting them.

  Language:Python777

• SamuelBrudner/juvenile_syllable_analysis

  Code to model juvenile syllable learning in latent space

  Language:Python4

• rtqichen/torchdiffeq

  Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.

  Language:Python5.7k940

• wsphillips/Conductor.jl

  Choo-choo

  Language:Julia6213

• fastplotlib/fastplotlib

  Next-gen fast plotting library running on WGPU using the pygfx rendering engine

  Language:Python42236

• PaulEcoffet/birdsynth

  Python module to generate bird song with the synthesizer from the DYNAMICAL SYSTEMS LAB Physics Department, Universidad de Buenos Aires

  Language:Jupyter Notebook32

• wsshin/jemdoc_mathjax

  jemdoc with MathJax support and more

  Language:Python23589

• gongziyida/POISEVAEv1

  Generic implementation

  Language:Jupyter Notebook1

• google-research/torchsde

  Differentiable SDE solvers with GPU support and efficient sensitivity analysis.

  Language:Python1.6k202