PettingZoo: Gym for Multi-Agent Reinforcement Learning
J. K. Terry, Benjamin Black, Nathaniel Grammel, Mario Jayakumar, Ananth Hari, Ryan Sullivan, Luis S. Santos, Clemens Dieffendahl, Caroline Horsch, Rodrigo Perez-Vicente, Niall L. Williams, Yashas Lokesh, Praveen Ravi
Abstract
This paper introduces the PettingZoo library and the accompanying Agent Environment Cycle ("AEC") games model. PettingZoo is a library of diverse sets of multi-agent environments with a universal, elegant Python API. PettingZoo was developed with the goal of accelerating research in Multi-Agent Reinforcement Learning ("MARL"), by making work more interchangeable, accessible and reproducible akin to what OpenAI's Gym library did for single-agent reinforcement learning. PettingZoo's API, while inheriting many features of Gym, is unique amongst MARL APIs in that it's based around the novel AEC games model. We argue, in part through case studies on major problems in popular MARL environments, that the popular game models are poor conceptual models of games commonly used in MARL and accordingly can promote confusing bugs that are hard to detect, and that the AEC games model addresses these problems.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers39
- Multi-agent Dynamic Algorithm ConfigurationKe Xue, Jiacheng Xu, Lei Yuan, Miqing Li et al.NeurIPS 2022 · 65 citations
- ACE: Cooperative Multi-Agent Q-learning with Bidirectional Action-DependencyChuming Li, Jie Liu, Yinmin Zhang, Yuhong Wei et al.AAAI 2023 · 38 citations
- A Stochastic Linearized Augmented Lagrangian Method for Decentralized Bilevel OptimizationSongtao Lu, Siliang Zeng, Xiaodong Cui, Mark S. Squillante et al.NeurIPS 2022 · 29 citations
- Latent Reward: LLM-Empowered Credit Assignment in Episodic Reinforcement LearningYun Qu, Yuhang Jiang, Boyuan Wang, Yixiu Mao et al.AAAI 2025 · 29 citations
- Scalable Primal-Dual Actor-Critic Method for Safe Multi-Agent RL with General UtilitiesDonghao Ying, Yunkai Zhang, Yuhao Ding, Alec Koppel et al.NeurIPS 2023 · 28 citations
Builds on1
Related papers
- Acting Beyond Learning: Imagination-Assisted Decision-Making in the Visual-based Multi-Agent Cooperative ScenariosHuanhuan Yang, Dianxi Shi, Songchang Jin, Guojun Xie et al.AAAI 2025
- UnrealZoo: Enriching Photo-Realistic Virtual Worlds for Embodied AIFangwei Zhong, Kui Wu, Churan Wang, Hao Chen et al.ICCV 2025 · 6 citations
- POPGym: Benchmarking Partially Observable Reinforcement LearningSteven D. Morad, Ryan Kortvelesy, Matteo Bettini, Stephan Liwicki et al.ICLR 2023 · 5 citations
- IntersectionZoo: Eco-driving for Benchmarking Multi-Agent Contextual Reinforcement LearningVindula Jayawardana, Baptiste Freydt, Ao Qu, Cameron Hickert et al.ICLR 2025
- Parallel Droplet Control in MEDA Biochips using Multi-Agent Reinforcement LearningTung-Che Liang, Jin Zhou, Yun-Sheng Chan, Tsung-Yi Ho et al.ICML 2021 · 15 citations
