JaxMARL: Multi-Agent RL Environments and Algorithms in JAX