JaxAHT: A JAX-Based Library for Ad Hoc Teamwork
JaxAHT is introduced, the first open-source, JAX-based library designed to accelerate and standardize the AHT research lifecycle and provides a unified framework for teammate generation, ego agent training, and evaluation against unseen teammates, achieving approximately 95x wall-clock speedup over PyTorch counterparts...