Brax is a fast and fully differentiable physics engine for large-scale rigid body simulations, built on JAX. It is designed for research in reinforcement learning and robotics, enabling efficient simulations and gradient-based optimization.

Features

  • GPU/TPU-accelerated physics simulations using JAX
  • Fully differentiable physics engine for gradient-based learning
  • Includes environments for RL benchmarks like humanoids and quadrupeds
  • Supports batched parallel simulations for high efficiency
  • Integrates seamlessly with reinforcement learning frameworks

Project Samples

Brax Screenshot 1

Project Activity

See All Activity >

Categories

Simulation, Physics, Reinforcement Learning Algorithms

License

Apache License V2.0

Follow Brax

Brax Web Site

Other Useful Business Software
Save Up to 91% on Cloud Compute With Spot VMs Icon
Save Up to 91% on Cloud Compute With Spot VMs

Automatic sustained-use discounts. One free VM per month. No negotiation needed.

Run batch jobs at 60-91% off with Spot VMs. Long-running workloads get automatic discounts with sustained use.
Start Free
Rate This Project
Login To Rate This Project

User Reviews

Be the first to post a review of Brax!

Additional Project Details

Programming Language

Python

Related Categories

Python Simulation Software, Python Physics Software, Python Reinforcement Learning Algorithms

Registered

2025-03-13
Report inappropriate content