jumanji (pypi): A diverse suite of scalable reinforcement learning environments in JAX
Keywords: reinforcement-learning|python|jax|research