JaxMARL: Multi-Agent RL Environments in JAX