Optional block-diffusion support for Qwen3 (attention + masked-diffusion SFT + on-policy distillation + RL rollout)#4516

ethannnnnn wants to merge 4 commits into

AI-Hypercomputer:mainAI-Hypercomputer/maxtext:mainfrom

ethannnnnn:dgr2-block-diffusionethannnnnn/maxtext:dgr2-block-diffusionCopy head branch name to clipboard