Skip to main content

Gym for multi-agent reinforcement learning

Project description

Build Status

PettingZoo is a Python library for conducting research in multi-agent reinforcement learning. It's akin to a multi-agent version of OpenAI's Gym library.

We model environments as Agent Environment Cycle (AEC) games, in order to be able to support all types of multi-agent RL environments under one API.

Environment Types and Installation

PettingZoo includes the following sets of games:

  • atari: Multi-player Atari 2600 games (both cooperative and competitive)
  • classic: Classical games including card games, board games, etc.
  • butterfly: Cooperative graphical games developed by us, requiring a high degree of coordination
  • magent: Configurable environments with massive numbers of particle agents, originally from https://github.com/geek-ai/MAgent
  • mpe: A set of simple nongraphical communication tasks, originally from https://github.com/openai/multiagent-particle-envs
  • sisl: 3 cooperative environments, originally from https://github.com/sisl/MADRL

To install, use pip install pettingzoo

We support Python 3.6, 3.7 and 3.8

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

PettingZoo-0.1.12.tar.gz (388.9 kB view hashes)

Uploaded Source

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page