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 Towards Top Cloud Services Icon
$300 in Free Credit Towards Top Cloud Services

Build VMs, containers, AI, databases, storage—all in one place.

Start your project in minutes. After credits run out, 20+ products include free monthly usage. Only pay when you're ready to scale.
Get Started
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