TABX: A High-Throughput Sandbox Battle Simulator for Multi-Agent Reinforcement Learning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Lee, Hayeong, Oh, JunHyeok, Lee, Byung-Jun
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910262458580992
author Lee, Hayeong
Oh, JunHyeok
Lee, Byung-Jun
author_facet Lee, Hayeong
Oh, JunHyeok
Lee, Byung-Jun
contents The design of environments plays a critical role in shaping the development and evaluation of cooperative multi-agent reinforcement learning (MARL) algorithms. While existing benchmarks highlight critical challenges, they often lack the modularity required to design custom evaluation scenarios. We introduce the Totally Accelerated Battle Simulator in JAX (TABX), a high-throughput sandbox designed for reconfigurable multi-agent tasks. TABX provides granular control over environmental parameters, permitting a systematic investigation into emergent agent behaviors and algorithmic trade-offs across a diverse spectrum of task complexities. Leveraging JAX for hardware-accelerated execution on GPUs, TABX enables massive parallelization and significantly reduces computational overhead. By providing a fast, extensible, and easily customized framework, TABX facilitates the study of MARL agents in complex structured domains and serves as a scalable foundation for future research. Our code is available at: https://github.com/ku-dmlab/TABX.
format Preprint
id arxiv_https___arxiv_org_abs_2602_01665
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle TABX: A High-Throughput Sandbox Battle Simulator for Multi-Agent Reinforcement Learning
Lee, Hayeong
Oh, JunHyeok
Lee, Byung-Jun
Multiagent Systems
Artificial Intelligence
Machine Learning
The design of environments plays a critical role in shaping the development and evaluation of cooperative multi-agent reinforcement learning (MARL) algorithms. While existing benchmarks highlight critical challenges, they often lack the modularity required to design custom evaluation scenarios. We introduce the Totally Accelerated Battle Simulator in JAX (TABX), a high-throughput sandbox designed for reconfigurable multi-agent tasks. TABX provides granular control over environmental parameters, permitting a systematic investigation into emergent agent behaviors and algorithmic trade-offs across a diverse spectrum of task complexities. Leveraging JAX for hardware-accelerated execution on GPUs, TABX enables massive parallelization and significantly reduces computational overhead. By providing a fast, extensible, and easily customized framework, TABX facilitates the study of MARL agents in complex structured domains and serves as a scalable foundation for future research. Our code is available at: https://github.com/ku-dmlab/TABX.
title TABX: A High-Throughput Sandbox Battle Simulator for Multi-Agent Reinforcement Learning
topic Multiagent Systems
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2602.01665