Jumanji: a Diverse Suite of Scalable Reinforcement Learning Environments in JAX

Clément Bonnet, Daniel Luo, Donal Byrne, Shikha Surana, Sasha Abramowitz +19 more
2/12/2026

No official code found. Best implementation: instadeepai/jumanji (828 stars, Apache-2.0, JAX, last commit 2026-04-13).

Semantic Scholar

Code Implementations(1)

instadeepai/jumanjiCommunity71%
JAXApache-2.0

🕹️ A diverse suite of scalable reinforcement learning environments in JAX

82895Python, JavaScriptAug 11, 20226 months agoApache-2.0
jaxpythonreinforcement-learningresearch

Cite this paper

@article{bonnet2026jumanji,
  title  = {Jumanji: a Diverse Suite of Scalable Reinforcement Learning Environments in JAX},
  author = {Clément Bonnet and Daniel Luo and Donal Byrne and Shikha Surana and Sasha Abramowitz and Paul Duckworth and Vincent Coyette and L. Midgley and Elshadai Tegegn and Tristan Kalloniatis and Omayma Mahjoub and Matthew Macfarlane and Andries P. Smit and Nathan Grinsztajn and Raphael Boige and Cemlyn N. Waters and Mohamed A. Mimouni and Ulrich A. Mbou Sob and Ruan de Kock and Siddarth Singh and Daniel Furelos-Blanco and Victor Le and Arnu Pretorius and Alexandre Laterre},
  year   = {2026},
  url    = {https://api.semanticscholar.org/CorpusID:7d1fb8ba683189d6d0fa2df0488bba2de1770355},
  journal = {ICLR 2024 2024}
}

Discussion