jit-env (pypi): A Jax interface for Reinforcement Learning environments.
Keywords: reinforcement-learning|python|machine|learning|jax