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