Massively parallel rigidbody physics simulation on accelerator hardware.
☆3,153May 4, 2026Updated 2 weeks ago
Alternatives and similar repositories for brax
Users that are interested in brax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- RL Environments in JAX 🌍☆894Apr 2, 2026Updated last month
- Tiny Differentiable Simulator is a header-only C++ and CUDA physics library for reinforcement learning and robotics with zero dependencie…☆1,361Oct 18, 2024Updated last year
- ☆1,426Mar 31, 2026Updated last month
- An open-source library for GPU-accelerated robot learning and sim-to-real transfer.☆1,942May 11, 2026Updated last week
- GPU-optimized version of the MuJoCo physics simulator, designed for NVIDIA hardware.☆1,229May 13, 2026Updated last week
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Google DeepMind's software stack for physics-based simulation and Reinforcement Learning environments, using MuJoCo.☆4,578May 11, 2026Updated last week
- C++-based high-performance parallel environment execution engine (vectorized env) for general RL environments.☆1,405May 12, 2026Updated last week
- Isaac Gym Reinforcement Learning Environments☆2,934Oct 26, 2024Updated last year
- Multi-Joint dynamics with Contact. A general purpose physics simulator.☆13,489Updated this week
- A collection of high-quality models for the MuJoCo physics engine, curated by Google DeepMind.☆3,441May 9, 2026Updated last week
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.☆756Oct 26, 2022Updated 3 years ago
- Real-time behaviour synthesis with MuJoCo, using Predictive Control☆1,636Apr 23, 2026Updated 3 weeks ago
- Evolution Strategies in JAX 🦎☆758Apr 2, 2026Updated last month
- RL implementations☆1,348May 4, 2026Updated 2 weeks ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Imitation learning benchmark focusing on complex locomotion tasks using MuJoCo.☆1,397May 2, 2026Updated 2 weeks ago
- Really Fast End-to-End Jax RL Implementations☆1,068Sep 9, 2024Updated last year
- Optax is a gradient processing and optimization library for JAX.☆2,269Updated this week
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms☆588May 11, 2026Updated last week
- 🕹️ A diverse suite of scalable reinforcement learning environments in JAX☆832Apr 13, 2026Updated last month
- ☆948Jun 27, 2024Updated last year
- A differentiable physics engine for robotics☆385Sep 23, 2024Updated last year
- Nimble: Physics Engine for Biomechanics and Deep Learning☆480Sep 14, 2025Updated 8 months ago
- High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, T…☆9,785Apr 20, 2026Updated last month
- End-to-end encrypted cloud storage - Proton Drive • AdSpecial offer: 40% Off Yearly / 80% Off First Month. Protect your most important files, photos, and documents from prying eyes.
- Unified framework for robot learning built on NVIDIA Isaac Sim☆7,160Updated this week
- ☆940Apr 10, 2026Updated last month
- [ICLR 2022] Accelerated Policy Learning with Parallel Differentiable Simulation☆360May 3, 2024Updated 2 years ago
- ♟️ Vectorized RL game environments in JAX☆605Mar 6, 2025Updated last year
- A library of reinforcement learning components and agents☆3,985Apr 8, 2026Updated last month
- A Python framework for GPU-accelerated simulation, robotics, and machine learning.☆6,666Updated this week
- JAX-based neural network library☆3,231May 12, 2026Updated last week
- Modular Reinforcement Learning (RL) library (implemented in PyTorch, JAX, and NVIDIA Warp) with support for Gymnasium/Gym, NVIDIA Isaac L…☆1,049May 11, 2026Updated last week
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/☆2,881May 11, 2026Updated last week
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Library for Model Based RL☆1,061Jul 12, 2024Updated last year
- Flax is a neural network library for JAX that is designed for flexibility.☆7,198May 12, 2026Updated last week
- ☆634May 13, 2026Updated last week
- PyTorch version of Stable Baselines, reliable implementations of reinforcement learning algorithms.☆13,260May 11, 2026Updated last week
- [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.☆872Aug 12, 2024Updated last year
- Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more☆35,627Updated this week
- A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.☆3,431Updated this week