Исследовательская группа Reactor представила Open Dreamer — полноценную открытую реализацию конвейера мировых моделей Dreamer 4. Проект, написанный на JAX и Flax NNX, включает в себя все ключевые компоненты для обучения агентов, работающих с видеотокенизацией и латентной динамикой, что позволяет сообществу воспроизводить сложные архитектуры обучения с подкреплением в контролируемой среде.

Релиз включает два репозитория, охватывающих полный цикл обучения: от каузального видеотокенизатора до модели латентной динамики, обусловленной действиями (action-conditioned). Использование стека JAX обеспечивает высокую производительность при вычислениях, что критически важно для обучения моделей, предсказывающих развитие состояний среды в долгосрочной перспективе. Публикация рецепта обучения дает исследователям инструмент для глубокого анализа того, как агенты строят внутренние репрезентации мира.

Данная реализация упрощает доступ к передовым методам обучения моделей мира, которые ранее были ограничены закрытыми проприетарными разработками. Наличие готовых инструментов для генерации роллаутов и оценки качества через FVD (Frechet Video Distance) позволяет стандартизировать процесс тестирования новых алгоритмов принятия решений и планирования в агентных системах.

Ключевые факты

  • Проект реализован на базе фреймворков JAX и Flax NNX.
  • В состав репозитория входят каузальный видеотокенизатор и модель латентной динамики.
  • Реализованы инструменты для генерации роллаутов и метрики FVD для оценки качества генерации.
  • Релиз включает полный рецепт обучения, обеспечивающий воспроизводимость результатов.
  • Разработчиком выступила исследовательская группа Reactor.