We present Brax, an open source library for rigid body simulation with a focus on performance and parallelism on accelerators. Brax reaches millions of simulation steps per second on environments like OpenAI Gym’s MuJoCo

2021: C. Freeman, Erik Frey, Anton Raichuk, Sertan Girgin, Igor Mordatch, Olivier Bachem

Keywords: Simulation

https://arxiv.org/pdf/2106.13281v1.pdf