JaxAHT library provides 95x speedup for Ad Hoc Teamwork research
September 15, 2026
JaxAHT is an open-source, JAX-based library designed to accelerate agent coordination research. It utilizes hardware acceleration to achieve a 95x wall-clock speedup over PyTorch implementations for teammate generation and evaluation across domains like Overcooked and Hanabi.
HOW THIS AFFECTS YOU
●
researcherYou can significantly reduce the training and evaluation latency of multi-agent coordination experiments.