Open Dreamer 发布:基于 JAX/Flax 的 Dreamer 4 世界模型开源复现,含完整训练配方

Meet Open Dreamer: A JAX/Flax Reproduction of the Dreamer 4 World Model Pipeline, With the Full Training Recipe Published

精选理由

Reactor 团队把 Dreamer 4 的完整训练配方和代码都开源了,JAX/Flax 实现,想复现世界模型做视频预测的赶紧看。

AI 摘要

Reactor 团队开源了 Open Dreamer,这是一个基于 JAX 和 Flax NNX 的 Dreamer 4 世界模型管道的完整复现。该项目包含两个代码仓库:next-state/open-dreamer 提供训练管道,包括因果视频分词器、动作条件潜在动力学模型、轨迹生成和 FVD 评分;reactor-team/open-dreamer 提供推理代码。Open Dreamer 的发布使 Dreamer 4 的完整训练流程得以公开复现。

图片来源 · marktechpost
原文 · marktechpost

Meet Open Dreamer: A JAX/Flax Reproduction of the Dreamer 4 World Model Pipeline, With the Full Training Recipe Published

A small group of AI researchers (Reactor) have released Open Dreamer, an open implementation of the Dreamer 4 world-model pipeline written in JAX and Flax NNX. What actually shipped Two repositories were released. next-state/open-dreamer holds the training pipeline: a causal video tokenizer, an action-conditioned latent dynamics model, rollout generation, and FVD scoring. reactor-team/open-dreamer holds a […] The post Meet Open Dreamer: A JAX/Flax Reproduction of the Dreamer 4 World Model Pipeline, With the Full Training Recipe Published appeared first on MarkTechPost .