brax (pypi): A differentiable physics engine written in JAX.
Keywords: JAX|reinforcement|learning|rigidbody|physics|jax|physics-simulation|reinforcement-learning|robotics