Abstract
Accurate economic simulations often require many experimental runs,particularly when combined with reinforcement learning. Unfortunately, trainingreinforcement learning agents in multi-agent economic environments can be slow.This paper introduces EconoJax, a fast simulated economy, based on the AIeconomist. EconoJax, and its training pipeline, are completely written in JAX.This allows EconoJax to scale to large population sizes and perform largeexperiments, while keeping training times within minutes. Through experimentswith populations of 100 agents, we show how real-world economic behavioremerges through training within 15 minutes, in contrast to previous work thatrequired several days. To aid and inspire researchers to build more rich anddynamic economic simulations, we open-source EconoJax on Github at:https://github.com/ponseko/econojax.