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 Free Credits to Build on Google Cloud Icon
$300 Free Credits to Build on Google Cloud

New customers can spin up VMs, build with AI, and query data at no cost.

Put your $300 in credit toward real workloads, then keep building with free monthly usage for 20+ products. No commitment and no charge until you upgrade.
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