Jumanji: a Diverse Suite of Scalable Reinforcement Learning Environments in JAX - Bonnet, Clem; Luo, Daniel; Byrne, Donal; Surana, Shikha; Abramowitz, Sasha; Duckworth, Paul; Coyette, Vincent; Midgley, Laurence; Tegegn, Elshadai; Kalloniatis, Tristan; Mahjoub, Omayma; Macfarlane, Matthew; et al. (International Conference on Learning Representations, 2024)