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