arXiv · ML· Koen Ponse, Aske Plaat, Thomas M. Moerland·· 5 小时前AI 评分30
Jaxnasium: 基于 PyTrees 的跨套件多智能体强化学习库
Jaxnasium: Embracing PyTrees For Any-Agent, Cross-Suite Reinforcement Learning
AI 导读
Jaxnasium 是一个基于 JAX 的开源强化学习库，利用 PyTrees 结构支持跨 11 个环境套件的 300 多个环境，允许在离散、连续、图像及组合空间中进行多智能体训练，无需修改算法代码。该库提供近乎单文件实现的 PPO 算法，默认超参数即可运行，并具备 Gymnasium/EnvPool 包装器、多种子评估、超参数扫描及项目脚手架工具。
来源：arXiv · ML · arxiv.org