跳到正文
arXiv Artificial Intelligence· Caroline Wang, Rolando Fernandez, Zelal Su Mustafaoglu, Montek Singh Kundan, Johnny Liu, Aditya Madhan, Zhihan Wang, Lingyun Xiao, Di Yang Shi, Jiaxun Cui, Arrasy Rahman, Peter Stone·· 19 小时前AI 评分22

JaxAHT:基于 JAX 的 Ad Hoc Teamwork 开源库

JaxAHT: A JAX-Based Library for Ad Hoc Teamwork

AI 导读

JaxAHT 是首个基于 JAX 的开源 Ad Hoc Teamwork(AHT)库,统一支持队友生成、ego agent 训练与未知队友评估,相比现有 PyTorch 实现取得约 95 倍墙钟加速。

来源:arXiv Artificial Intelligence · arxiv.org