← all repositories

google/brax

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

3.2k stars Jupyter Notebook ML FrameworksAgentsDomain Apps
brax
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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.