arXiv 机器学习· Koen Ponse, Aske Plaat, Thomas M. Moerland·· 6 小时前AI 评分50
Jaxnasium：利用 PyTrees 实现任意智能体与跨套件强化学习
Jaxnasium: Embracing PyTrees For Any-Agent, Cross-Suite Reinforcement Learning
AI 导读
基于 JAX 的强化学习库 Jaxnasium 正式开源，该项目将观测与动作空间表示为 PyTrees，无需修改算法代码即可适配离散、连续、图像及多智能体环境。该库提供极简 Gymnasium 风格接口与 11 个套件封装，开箱支持超 300 个环境及小型 PPO 实现，同时可无损兼容 Gymnasium 和 EnvPool。此外项目还提供多种子评估与超参数搜索工具。
来源：arXiv 机器学习 · arxiv.org