DiffusionGemma: 1,000+ tokens/sec open-weights text gen

google-deepmindgemmadiffusiongemmatext-diffusionopen-weightsapache-2-0+9high-risk claims
DiffusionGemma social card showing the text-diffusion approach with a Gemma 4 base and parallel token generation
Image: Google blog / DiffusionGemma announcement (June 10, 2026)

On June 10, 2026, Google DeepMind released DiffusionGemma, an open-weights model that abandons the single-token autoregressive (AR) pattern and generates text via diffusion: it denoises 256-token blocks in parallel, reaching 1,000+ tokens/sec on a single H100 — roughly 4x faster than an equivalent AR model in single-user mode (Google blog, June 10, 2026; NVIDIA blog, June 10, 2026). It’s a 26B total / 3.8B active MoE based on Gemma 4, released under Apache 2.0 with day-one support in vLLM, Hugging Face Transformers, Unsloth, NVIDIA NeMo, and TensorRT-LLM with NVFP4 on Blackwell GPUs. Google explicitly labels it “experimental”: quality is below Gemma 4 AR — speed is the point, not peak quality.

What was released

The post on Google’s official blog, authored by Brendan O’Donoghue and Sebastian Flennerhag, frames DiffusionGemma as an “experimental open model that explores text diffusion, an exceptionally fast approach to text generation.” Weights are on Hugging Face under google/diffusiongemma-26B-A4B-it with apache-2.0. A Blackwell-optimized NVFP4 build is published by NVIDIA.

The architecture inherits the Gemma 4 26B A4B MoE backbone (8 of 128 experts active, 25.2B total / 3.8B active per step) and adds a diffusion head that denoises 256-token blocks in parallel. Attention is no longer purely causal — every token attends to all others. Context reaches 256K tokens with a 1,024-token sliding window. Inputs are multimodal (text, images, video); outputs are text-only.

Speed and benchmarks

HardwareTokens/secNotes
NVIDIA H100 (single, FP8, low batch)1,000+HF card: “exceeding 1100 tokens/sec”
NVIDIA RTX 5090 (quantized)700+Google blog
NVIDIA DGX Station2,000NVIDIA blog
NVIDIA DGX Spark (GB10)150NVIDIA blog

The model card compares DiffusionGemma 26B A4B (instruction-tuned, Entropy Bound sampler) against Gemma 4 26B A4B AR — the deltas are honest: −5.0 pp on MMLU Pro, −19.2 pp on AIME 2026, −8.0 pp on LiveCodeBench v6, −289 ELO on Codeforces, −12.1 pp on MRCR v2, +0.170 (worse) on OmniDocBench 1.5. The single positive delta is HLE no tools (+2.3 pp) at low absolute scores. Google frames the speedup as “up to 4x faster”: “The throughput advantage is strongest at low-to-medium batch sizes on a single accelerator.”

Why it matters

Day-one ecosystem

Integrations: Hugging Face Transformers (DiffusionGemmaForBlockDiffusion), vLLM (day-zero, Red Hat AI support), MLX, Unsloth, NVIDIA NeMo, and TensorRT-LLM with NVFP4 on Blackwell. Cloud paths: Gemini Enterprise Model Garden and NVIDIA NIM. llama.cpp support is “coming soon” — that would open Mac.

Practical advice

1. IDE / editor completion teams. Profile latency to first block (256 tokens), not first token. Bidirectional attention refines tokens AR would have frozen — useful for non-linear edits. 2. Agent harness builders. Read the DiffusionGemmaForBlockDiffusion implementation and vLLM day-zero PR. Block-autoregressive decoding changes the KV-cache story. 3. Benchmark evaluators. Re-run the deltas independently — contribution is speed, not a new SOTA. Deltas vs Gemma 4 AR: −5.0 MMLU Pro, −19.2 AIME, −8.0 LiveCodeBench, −289 ELO Codeforces, −12.1 MRCR v2, +0.170 OmniDocBench, +2.3 HLE (pp). 4. Local inference users. Measure VRAM: 18 GB (Google) vs 24 GB (DeepMind) — difference is quantization and context length. 5. Cloud serving architects. Google: speedup “collapses in high-QPS cloud serving.” Use DiffusionGemma for low-to-medium batch single-user; keep AR for batched serving.

What to watch

  1. Independent benchmarks on SWE-bench, LiveCodeBench, GPQA, AIME. Numbers above are from Google’s model card, not independent runs.
  2. vLLM and Hugging Face TGI serving stability for block-autoregressive decoding with KV-cache management.
  3. Blackwell NVFP4 quality samples. NVIDIA calls it “near-lossless” — measure on your workload before production.
  4. llama.cpp support for non-NVIDIA hardware.
  5. Fine-tuning lift on vertical use cases (code infilling, structured extraction, IDE completion).
  6. Independent safety benchmarks (HarmBench, AdvBench, third-party red teams).

Risks and caveats

Verdict

On June 10, 2026, Google DeepMind released DiffusionGemma: 26B/3.8B active, Gemma 4 MoE, 256 tokens denoised in parallel, 1,000+ tokens/sec on a single H100, ~4x faster than equivalent AR in single-user mode, Apache 2.0, day-one support in vLLM, Hugging Face, Unsloth, NeMo, TensorRT-LLM with NVFP4 on Blackwell. It’s the first open-weights text-diffusion model at this scale — Google labels it “experimental”, with quality systematically below Gemma 4 AR. Where end-to-end latency matters — IDE completion, code infilling, single-user agent loops, local inference on consumer GPUs — DiffusionGemma is a serious candidate.

Sources