SimulatorGoogle · 2021

Brax

JAX-based, massively-parallel differentiable physics engine for reinforcement learning on GPU/TPU.

Summary

Brax is a differentiable physics engine written in JAX that simulates thousands of rigid-body environments in parallel on a single accelerator, making it a fast substrate for RL and evolutionary training. Apache-2.0.

Metadata

Organization: Google
Year: 2021
License: Apache-2.0
Backend: JAX (GPU/TPU)
Differentiable: Yes

Relationships

Competes with

MuJoCo

#differentiable#jax#rl