marl-jax: Multi-agent Reinforcement Leaning framework for Social Generalization

03/24/2023
by   Kinal Mehta, et al.
0

Recent advances in Reinforcement Learning (RL) have led to many exciting applications. These advancements have been driven by improvements in both algorithms and engineering, which have resulted in faster training of RL agents. We present marl-jax, a multi-agent reinforcement learning software package for training and evaluating social generalization of the agents. The package is designed for training a population of agents in multi-agent environments and evaluating their ability to generalize to diverse background agents. It is built on top of DeepMind's JAX ecosystem~\cite{deepmind2020jax} and leverages the RL ecosystem developed by DeepMind. Our framework marl-jax is capable of working in cooperative and competitive, simultaneous-acting environments with multiple agents. The package offers an intuitive and user-friendly command-line interface for training a population and evaluating its generalization capabilities. In conclusion, marl-jax provides a valuable resource for researchers interested in exploring social generalization in the context of MARL. The open-source code for marl-jax is available at: \href{https://github.com/kinalmehta/marl-jax}{https://github.com/kinalmehta/marl-jax}

READ FULL TEXT
research
12/13/2021

PantheonRL: A MARL Library for Dynamic Training Interactions

We present PantheonRL, a multiagent reinforcement learning software pack...
research
11/28/2022

Improved Representation of Asymmetrical Distances with Interval Quasimetric Embeddings

Asymmetrical distance structures (quasimetrics) are ubiquitous in our li...
research
06/18/2020

Efficient Ridesharing Dispatch Using Multi-Agent Reinforcement Learning

With the advent of ride-sharing services, there is a huge increase in th...
research
10/07/2020

Pkwrap: a PyTorch Package for LF-MMI Training of Acoustic Models

We present a simple wrapper that is useful to train acoustic models in P...
research
07/03/2019

Reasoning and Generalization in RL: A Tool Use Perspective

Learning to use tools to solve a variety of tasks is an innate ability o...
research
06/11/2023

Herd's Eye View: Improving Game AI Agent Learning with Collaborative Perception

We present a novel perception model named Herd's Eye View (HEV) that ado...
research
10/01/2020

Learning Social Learning

Social learning is a key component of human and animal intelligence. By ...

Please sign up or login with your details

Forgot password? Click here to reset