Open Dreamer: JAX/Flax Reproduction of the Dreamer 4 World Model Pipeline
Open Dreamer showcases a JAX/Flax reproduction of the Dreamer 4 world model pipeline with the full training recipe.
Open Dreamer presents a JAX/Flax reproduction of the Dreamer 4 world model pipeline. It utilizes two models sharing a block-causal transformer: a video tokenizer and an action-conditioned dynamics model. Masking allows for a more diffusible latent space, eliminating the need for KL or adversarial loss.
This synthesis was produced from its source by AI; there is no human editor or manual review step. How we work