Sourcify
Effortlessly find and manage open-source dependencies for your projects.

Massively parallel rigidbody physics simulation on accelerator hardware.

Brax is a fast and fully differentiable physics engine written in JAX, designed for research and development in robotics, human perception, materials science, and reinforcement learning. It supports efficient single-device simulation and massively parallel simulation on multiple devices, particularly excelling on TPUs, achieving millions of physics steps per second. Brax offers four physics pipelines: MuJoCo XLA (MJX), Generalized coordinates, Positional dynamics, and Spring-based simulation. It includes baseline learning algorithms like PPO, SAC, ARS, and analytic policy gradients. Brax facilitates experiments in transfer learning and bridging the simulation-to-real-world gap, offering an API for seamless pipeline swapping. While primarily an RL library, it encourages users to adopt MJX or MuJoCo Warp for pure physics simulations.
Brax is a fast and fully differentiable physics engine written in JAX, designed for research and development in robotics, human perception, materials science, and reinforcement learning.
Explore all tools that specialize in reinforcement learning. This domain focus ensures Brax delivers optimized results for this specific requirement.
Brax is fully differentiable, allowing for the calculation of gradients through the simulation for tasks like analytic policy gradients.
Designed for parallel execution on accelerators like TPUs, enabling millions of physics steps per second.
Offers four distinct physics pipelines (MJX, Generalized, Positional, Spring) that can be easily swapped within the same simulation.
Written in JAX, which provides automatic differentiation, GPU/TPU acceleration, and composable function transformations.
Includes baseline learning algorithms such as PPO, SAC, ARS, and evolutionary strategies.
Install Brax from PyPI: `pip install brax`
Alternatively, install from Conda: `conda install -c conda-forge brax`
For GPU support, install CUDA, CuDNN, and JAX with GPU support.
Explore the Brax API using the Brax Basics Colab notebook.
Train policies with Brax Training Colab notebook.
Experiment with different physics pipelines (MJX, Generalized, Positional, Spring).
Consult the documentation for advanced usage and customization.
All Set
Ready to go
Verified feedback from other users.
"Brax is highly regarded for its speed, differentiability, and scalability in physics simulation and reinforcement learning research."
Post questions, share tips, and help other users.
Effortlessly find and manage open-source dependencies for your projects.

End-to-end typesafe APIs made easy.

Page speed monitoring with Lighthouse, focusing on user experience metrics and data visualization.

Topcoder is a pioneer in crowdsourcing, connecting businesses with a global talent network to solve technical challenges.

Explore millions of Discord Bots and Discord Apps.

Build internal tools 10x faster with an open-source low-code platform.

Open-source RAG evaluation tool for assessing accuracy, context quality, and latency of RAG systems.

AI-powered synthetic data generation for software and AI development, ensuring compliance and accelerating engineering velocity.