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
The AI workplace management platform Icon
The AI workplace management platform

Plan smart spaces, connect teams, manage assets, and get insights with the leading AI-powered operating system for the built world.

By combining AI workflows, predictive intelligence, and automated insights, OfficeSpace gives leaders a complete view of how their spaces are used and how people work. Facilities, IT, HR, and Real Estate teams use OfficeSpace to optimize space utilization, enhance employee experience, and reduce portfolio costs with precision.
Learn More
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