Factoriax: A GPU-Accelerated Factorio-Style Simulator for Reinforcement Learning

Mickey Beurskens, Tristan Tomilin, Thiago D. Simão

We introduce Factoriax, a GPU-accelerated factory-building simulator written in JAX. In Factoriax, an agent must collect resources, build machines using those resources, and then a…