跳到正文
事件追踪已归档

Jaxnasium:用PyTrees统一JAX强化学习环境

首次报道 最新进展 1 篇报道,1 个来源

事件综述根据 1 篇报道整理,10月28日更新

Koen Ponse等作者发布开源JAX强化学习库Jaxnasium,把观测与动作空间表示为PyTrees,使接近单文件的算法实现无需改动即可运行于离散、连续、图像、复合及多智能体环境。 该库构建在Equinox上,提供Gymnasium风格的最小接口,封装11个套件、超过300个环境;作者称一个小型PPO实现用默认超参数即可运行。库还附带多随机种子评估、超参数扫描与项目脚手架工具,代码已在GitHub开源。 上述能力来自作者的预印本介绍,尚未见独立复现或第三方评测。

综述由模型根据报道生成,细节以各篇原文为准。

进展时间线

北京时间,最早在上
  1. arXiv cs.DL 数字图书馆Jaxnasium:基于PyTrees的JAX强化学习库

    阅读这篇报道

全部报道

1 篇,点开看摘要
Jaxnasium:用 PyTrees 统一任意智能体与跨环境套件的强化学习库arXiv cs.DL 数字图书馆预印本未入选

Jaxnasium 是构建在 Equinox 上的轻量 JAX 强化学习库,把观测与动作空间表示为 PyTrees,使近单文件算法实现无需改动即可运行于离散、连续、图像、复合及多智能体环境。它提供 Gymnasium 风格最小接口并封装 11 个套件、超过 300 个环境,小型 PPO 实现用默认超参数即可运行。该库已开源,还提供多随机种子评估、超参数扫描与项目脚手架工具。

阅读这篇报道