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

ethannnnnn wants to merge 6 commits into

AI-Hypercomputer:mainAI-Hypercomputer/maxtext:mainfrom

ethannnnnn:block-diffusion-maxtext-pr6-exact-replayethannnnnn/maxtext:block-diffusion-maxtext-pr6-exact-replayCopy head branch name to clipboard