google/brax
A fast, fully differentiable physics engine written in JAX for accelerating reinforcement learning and robotics research.

Velocity · 7d
+1.7
★ / day
Trend
→steady
star history
Brax provides massively parallel rigidbody physics simulation on accelerator hardware (TPU/GPU). Written entirely in JAX, it enables efficient single-device and distributed simulation without datacenter requirements. The library includes baseline RL algorithms (PPO, SAC, ARS) that can train agents in seconds to minutes, and is specifically designed for robotics, human perception, and reinforcement learning research applications.