Farama-Foundation/MPE2

A set of communication oriented environments

What it solves

MPE2 provides a set of communication-oriented simulation environments for training and testing multi-agent reinforcement learning (MARL) algorithms. It addresses the need for standardized, lightweight environments where agents must learn to interact, communicate, and coordinate to solve tasks in both cooperative and adversarial settings.

How it works

The project implements a series of "particle" environments where agents are represented as circles that can move, see other agents, and interact with static landmarks. It supports both discrete (default) and continuous action spaces, allowing agents to control their movement in cardinal directions or broadcast messages over communication channels. The system tracks observations (position, velocity, and received messages) and rewards based on the agents' success in achieving goals, such as pushing objects or tagging other agents.

Who it’s for

It is designed for researchers and developers working on multi-agent AI, specifically those focusing on emergent communication, coordination, and competitive/cooperative game theory in reinforcement learning.

Highlights

  • Diverse Interaction Models: Includes both cooperative environments (e.g., Simple Spread, Simple Formation) and adversarial ones (e.g., Simple Tag, Simple Push).
  • Flexible Action Spaces: Supports both discrete and continuous control for movement and communication.
  • Communication Channels: Built-in mechanisms for agents to broadcast and receive messages to coordinate actions.
  • Improved Stability: Based on OpenAI's original MPE, but with fixes for reward consistency and discrete action space defaults.

Related

  • Project
  • Project
  • Project
  • Dispatch
  • Project