Sign in

JaxMARL: Multi-Agent RL Environments in JAX

By Alexander Rutherford and others at
LogoRutherford University
and
LogoUniversity of Oxford
Benchmarks play an important role in the development of machine learning algorithms. For example, research in reinforcement learning (RL) has been heavily influenced by available environments and benchmarks. However, RL environments are traditionally run on the CPU, limiting their scalability with typical academic compute. Recent advancements in JAX have enabled... Show more
December 19, 2023
=
0
Loading PDF…
Loading full text...
Similar articles
Loading recommendations...
=
0
x1
JaxMARL: Multi-Agent RL Environments in JAX
Click on play to start listening