Multi-Agent Reinforcement Learning with JAX
Contributions:11 reviews, 42 PRs, 109 pushes in 5 months
jaxmulti-agent-reinforcement-learning
VMAS is a vectorized framework designed for efficient Multi-Agent Reinforcement Learning benchmarking. It is comprised of a vectorized 2D physics engine written in PyTorch and a set of challenging multi-robot scenarios. Additional scenarios can be implemented through a simple and modular interface.
Contributions:21 reviews, 27 PRs, 52 pushes in 7 months