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...
Caroline Wang, Rolando Fernandez, Z. Mustafaoglu et al.
· 2 citations