GitHub repository: Kinetix-Env/Jax2D
Description: Jax2D is a highly scalable 2D physics engine written in JAX.