[控场AI]
产品

PureJaxRL

基于JAX的强化学习库,利用jax.vmap和jax.lax.scan将整个RL训练循环表达为纯函数,在GPU上并行执行数千个环境实例

时间轴 (近 90 天)

10月16日

CARL受到PureJaxRL的启发,PureJaxRL曾在JAX上实现了类似的大规模并行环境方案。

待验证50%
10月16日

大规模并行环境是当前强化学习研究的一个重要方向,PureJaxRL、Isaac Gym等工作都在推动这一趋势。

待验证50%
10月16日

PureJaxRL 是一个基于 JAX 的强化学习库,利用 jax.vmap 和 jax.lax.scan 等原语将整个 RL 训练循环表达为纯函数,从而在 GPU 上并行执行数千个环境实例。

待验证50%

全部知识事实 (3)

来源文章