d-brax-jumpy (pypi): Common backend for JAX or numpy.
Keywords: jax|physics-simulation|reinforcement-learning|robotics