Shypd.aiThe largest open index of AI toolsJumanji is an open-source AI Frameworks & Infra tool that provides a diverse suite of scalable reinforcement learning environments in JAX. It enables faster iteration and large-scale experimentation for RL research.
Trending
About
Jumanji is an open-source library offering a diverse suite of scalable reinforcement learning environments, all written in JAX. It features 22 environments, ranging from simple games like Snake and Game2048 to complex NP-hard combinatorial problems such as BinPack and TSP. The tool is designed to accelerate hardware-accelerated research and development in RL by providing high-speed environments that facilitate faster iteration and large-scale experimentation while reducing complexity. Jumanji offers a simple, well-tested API for JAX-based environments, wrappers for popular RL frameworks like Acme and Gymnasium, and examples to guide adoption. It supports Python 3.10, 3.11, and 3.12, and leverages JAX features like automatic vectorization and JIT-compilation.
Capabilities
Pricing & Plans
Open Source
Free
FAQs
Trending
