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

Project Activity

See All Activity >

License

Apache License V2.0

Follow Brax

Brax Web Site

Other Useful Business Software
$300 in Free Credit for Your Google Cloud Projects Icon
$300 in Free Credit for Your Google Cloud Projects

Build, test, and explore on Google Cloud with $300 in free credit. No hidden charges. No surprise bills.

Launch your next project with $300 in free Google Cloud credit—no hidden charges. Test, build, and deploy without risk. Use your credit across the Google Cloud platform to find what works best for your needs. After your credits are used, continue building with free monthly usage products. Only pay when you're ready to scale. Sign up in minutes and start exploring.
Start Free Trial
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