Brax

Differentiable, massively parallel physics simulation in JAX. · By Google

Brax is a fast physics engine for RL, robotics and control research that runs on accelerators and supports MJX, positional and spring backends.