Add DiffusionGemma (text): block-diffusion model + diffusion_generate#1402

Draft

gilbert-barajas wants to merge 6 commits into

ml-explore:mainfrom

gilbert-barajas:diffusion-gemma-port

Commits on Jun 13, 2026

WIP diffusion_gemma: ModelArgs + slice 1 building blocks

gilbert-barajas
and
claude
committed
diffusion_gemma: full encoder-decoder model forwards end-to-end

gilbert-barajas
and
claude
committed
diffusion_gemma: diffusion_generate sampler — the port generates end-to-end

gilbert-barajas
and
claude
committed
diffusion_gemma: fix sampler/encoder bugs found by adversarial verification

authored and
gilbert-barajas
committed
diffusion_gemma: ModelArgs.from_dict for the nested mlx-community config

gilbert-barajas
and
claude
committed

Commits on Jun 14, 2026

diffusion_gemma: CLI wiring — mlx_lm.generate dispatches to diffusion_generate

gilbert-barajas
and
claude
committed