gymnax
JSON →gymnax provides JAX-ready implementations of classic reinforcement learning environments (e.g., CartPole, Acrobot, MinAtar) with fully functional and batched vectorization. Current version 0.0.9, with periodic releases.
gymnax provides JAX-ready implementations of classic reinforcement learning environments (e.g., CartPole, Acrobot, MinAtar) with fully functional and batched vectorization. Current version 0.0.9, with periodic releases.