EconoJax: A Fast & Scalable Economic Simulation in Jax
Koen Ponse, Aske Plaat, Niki van Stein, Thomas M. Moerland

TL;DR
EconoJax is a fast, scalable economic simulation framework built in JAX that enables rapid training of reinforcement learning agents in multi-agent economic environments, significantly reducing training time.
Contribution
The paper introduces EconoJax, a JAX-based economic simulation platform that accelerates training and experimentation in multi-agent economic models.
Findings
Training with EconoJax takes minutes instead of days.
Real-world economic behaviors emerge within 15 minutes of training.
Different multi-agent methods do not significantly affect behavioral diversity.
Abstract
Accurate economic simulations often require many experimental runs, particularly when combined with reinforcement learning. Unfortunately, training reinforcement learning agents in multi-agent economic environments can be slow. This paper introduces EconoJax, a fast simulated economy, based on the AI economist. EconoJax, and its training pipeline, are completely written in JAX. This allows EconoJax to scale to large population sizes and perform large experiments, while keeping training times within minutes. Through experiments with populations of 100 agents, we show how real-world economic behavior emerges through training within 15 minutes, in contrast to previous work that required several days. We additionally perform experiments in varying sized action spaces to test if some multi-agent methods produce more diverse behavior compared to others. Here, our findings indicate no notable…
Peer Reviews
No public reviews on file for this paper yet. If you reviewed it on a platform where reviews are public (OpenReview, ICLR, NeurIPS, ICML), you can paste yours below so the community can read it here.
Code & Models
Videos
No videos yet. Explain this paper in a talk, walkthrough, or lecture? Add one.
Taxonomy
TopicsDistributed and Parallel Computing Systems
