Open Dreamer veröffentlicht: Offene Implementierung der Dreamer-4-Pipeline in JAX/Flax mit vollständigem Trainingsrezept
Die KI-Forschungsgruppe Reactor hat Open Dreamer veröffentlicht – eine offene Implementierung der Dreamer-4-Weltmodell-Pipeline, geschrieben in JAX und Flax NNX. Die Repositories veröffentlichen den Code der Trainingspipeline (Video-Tokenizer, latentes Dynamikmodell, Trajektoriengenerierung und FVD-Bewertung) sowie zusätzliche Komponenten.
Eine kleine KI-Forschergruppe namens Reactor hat Open Dreamer veröffentlicht – eine offene Implementierung der Weltmodell-Pipeline Dreamer 4, geschrieben in JAX und Flax NNX. Es wurden zwei Repositorien veröffentlicht. Das Repositorium next-state/open-dreamer enthält die Trainingspipeline: einen kausalen Videotokenizer, ein latentes Dynamikmodell unter Berücksichtigung von Aktionen, die Generierung von Trajektorien und die FVD-Bewertung (Fréchet Video Distance). Das zweite Repositorium reactor-team/open-dreamer umfasst zusätzliche Komponenten.
Quelle: MarkTechPost —
Original