Jumanji is a diverse suite of scalable reinforcement learning environments written in JAX with 22+...