
NVIDIA Isaac Gym
A GPU-accelerated physics simulation environment for reinforcement learning research, now deprecated and succeeded by Isaac Lab.

Massively parallel rigidbody physics simulation on accelerator hardware.
Massively parallel rigidbody physics simulation on accelerator hardware.
Brax is a fast and fully differentiable physics engine written in JAX, designed for research and development in robotics, human perception, materials science, and reinforcement learning. It supports efficient single-device simulation and massively parallel simulation on multiple devices, particularly excelling on TPUs, achieving millions of physics steps per second. Brax offers four physics pipelines: MuJoCo XLA (MJX), Generalized coordinates, Positional dynamics, and Spring-based simulation. It includes baseline learning algorithms like PPO, SAC, ARS, and analytic policy gradients. Brax facilitates experiments in transfer learning and bridging the simulation-to-real-world gap, offering an API for seamless pipeline swapping. While primarily an RL library, it encourages users to adopt MJX or MuJoCo Warp for pure physics simulations.
Massively parallel rigidbody physics simulation on accelerator hardware.
Quick visual proof for Brax. Helps non-technical users understand the interface faster.
Brax is a fast and fully differentiable physics engine written in JAX, designed for research and development in robotics, human perception, materials science, and reinforcement learning.
Explore all tools that specialize in reinforcement learning. This domain focus ensures Brax delivers optimized results for this specific requirement.
Open side-by-side comparison first, then move to deeper alternatives guidance.
Brax is fully differentiable, allowing for the calculation of gradients through the simulation for tasks like analytic policy gradients.
Designed for parallel execution on accelerators like TPUs, enabling millions of physics steps per second.
Offers four distinct physics pipelines (MJX, Generalized, Positional, Spring) that can be easily swapped within the same simulation.
Written in JAX, which provides automatic differentiation, GPU/TPU acceleration, and composable function transformations.
Includes baseline learning algorithms such as PPO, SAC, ARS, and evolutionary strategies.
Install Brax from PyPI: `pip install brax`
Alternatively, install from Conda: `conda install -c conda-forge brax`
For GPU support, install CUDA, CuDNN, and JAX with GPU support.
Explore the Brax API using the Brax Basics Colab notebook.
Train policies with Brax Training Colab notebook.
Experiment with different physics pipelines (MJX, Generalized, Positional, Spring).
Consult the documentation for advanced usage and customization.
All Set
Ready to go
Verified feedback from other users.
“Brax is highly regarded for its speed, differentiability, and scalability in physics simulation and reinforcement learning research.”
No reviews yet. Be the first to rate this tool.

A GPU-accelerated physics simulation environment for reinforcement learning research, now deprecated and succeeded by Isaac Lab.

Ray is an open-source AI compute engine for scaling AI and Python applications.

An open-source library for reinforcement learning, offering scalable and fault-tolerant RL workloads.

Powering AI with expert human data to build, evaluate, and align frontier AI.

Uncover and optimize your SaaS investment.

A powerful shell designed for interactive use and scripting.