GitHub repository: moritzf/brax
Description: Massively parallel rigidbody physics simulation on accelerator hardware.