instadeepai/jumanji
A scalable suite of reinforcement learning environments implemented in JAX for combinatorial optimization and research tasks.

Velocity · 7d
+0.6
★ / day
Trend
→steady
star history
Jumanji provides a collection of reinforcement learning environments written in JAX, enabling fast simulation and training of RL agents. It includes environments for combinatorial optimization problems such as bin packing, vehicle routing, and connector assembly. The project integrates with Hugging Face for sharing trained agents and is designed for scalable RL research.