[maxtext] Add exact block-diffusion policy replay - #4591

Open

ethannnnnn wants to merge 6 commits into

AI-Hypercomputer:mainfrom

ethannnnnn:block-diffusion-maxtext-pr6-exact-replay

Commits on Jul 31, 2026

[maxtext] Add generalized block-causal attention
ethannnnnn
committed
ae07067
[maxtext] Add generalized block-diffusion CFT and SFT
ethannnnnn
committed
ab7bb88
[maxtext] Add reusable block-diffusion rollout transitions
ethannnnnn
committed
2b5703d
[maxtext] Add model-aware block-diffusion OPD
ethannnnnn
committed
b32cdd6
[maxtext] Export exact stochastic denoising traces
ethannnnnn
committed
70ebc7c
[maxtext] Add exact block-diffusion policy replay
ethannnnnn
committed
3a3aa3c