← all repositories

instadeepai/jumanji

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

840 stars Python ML FrameworksDomain Apps
jumanji
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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.