Jumanji is a diverse suite of scalable reinforcement learning environments written in JAX, featuring...