Skip to content
#

pmpo

Star

Here is 1 public repository matching this topic...

vijayabhaskar-ev / dreamer_v4

Star 22

From-scratch PyTorch implementation of DreamerV4 (Hafner et al., 2024): masked-autoencoder tokenizer, block-causal flow-matching dynamics with bootstrap curriculum, agent-token finetuning, and PMPO imagination RL. Hardened for TPU v4 / torch_xla with fixed-shape graphs, on-device RNG, and bounded compile-cache footprint.

reinforcement-learning pytorch transformer tpu model-based-rl dreamer world-models world-model flow-matching dreamer-v4 torch-xla pmpo
  • Updated Jun 5, 2026
  • Python

Improve this page

Add a description, image, and links to the pmpo topic page so that developers can more easily learn about it.

Curate this topic

Add this topic to your repo

To associate your repository with the pmpo topic, visit your repo's landing page and select "manage topics."

Learn more