Mava

Mava 

A research-friendly codebase for fast experimentation of multi-agent reinforcement learning in JAX.