Shypd.aiThe largest open index of AI toolsgymnax is a reinforcement learning library built on JAX and Haiku. It provides a gym-like API for various RL environments, enabling high-throughput experiments through massive vectorization.
Trending
About
gymnax is a reinforcement learning library designed to leverage the capabilities of JAX and Haiku. It offers an API similar to the classic gym library, making it familiar for researchers and developers in the RL domain. The library supports a diverse set of environments, including classic control problems, bsuite, MinAtar, and both classic and meta-RL tasks. A key feature of gymnax is its ability to facilitate massive vectorization, which significantly boosts throughput for reinforcement learning experiments.
Capabilities
Pricing & Plans
unknown
Free
FAQs
Trending
