Tuesday, July 7, 2026

Diffusion LLMs Explained: Inception Labs' Mercury 2, Google's DiffusionGemma, and the Open-Source Race

Every major chatbot you've used — ChatGPT, Claude, Gemini — generates text the same way: one token at a time, left to right, like a typewriter. Diffusion language models throw that out. They sketch a rough draft all at once and refine it in parallel, the same way image generators turn static into a photo. Here's what that actually means, who's building these models, and where they fall short.

How Diffusion LLMs Actually Work

Standard "autoregressive" LLMs predict the next word based on everything written so far, one token after another. It works well, but it's fundamentally sequential — you can't generate token 500 until you've generated token 499.

Diffusion LLMs borrow the idea behind image generators like Stable Diffusion: start with something close to random noise (in this case, a block of placeholder or masked tokens) and iteratively refine the entire block in parallel across multiple passes, with every token able to "see" every other token at once. Inception Labs' CEO describes it as the difference between a typewriter and an editor revising a full draft simultaneously. Because this refinement process is parallel rather than sequential, it maps naturally onto what GPUs are actually good at: doing lots of math at once, rather than shuffling data through memory one step at a time.

Watch: The Core Idea Behind Diffusion LLMs

Inception Labs: Mercury and Mercury 2

Inception Labs, founded by Stanford professor Stefano Ermon — who helped pioneer diffusion modeling for images and co-authored the foundational text-diffusion research — introduced the original Mercury family as the first commercial-scale diffusion LLMs, including a code-focused variant called Mercury Coder.

The current flagship is Mercury 2, released in February 2026 and positioned as the first reasoning-capable diffusion LLM. Key specs and claims:

  • Over 1,000 output tokens per second on Nvidia Blackwell GPUs — roughly 5–10x faster end-to-end than speed-optimized models from OpenAI, Anthropic, and Google.
  • 128K context window, tunable reasoning depth, native tool use, and schema-aligned JSON output.
  • OpenAI-compatible API, marketed as a drop-in replacement requiring no rewrites.
  • Pricing around $0.25 per million input tokens and $0.75 per million output tokens.

On quality, Ermon has been candid: Mercury 2 targets and matches models like Claude Haiku and Gemini Flash-class models, not frontier flagships like Claude Opus or GPT-4-class reasoning models. The pitch isn't "smartest model available" — it's "reasoning-grade quality at a fraction of the latency and cost," which matters most for real-time voice agents, high-volume search/retrieval pipelines, and agent loops where latency compounds across many steps.

Watch: Inception's Approach in Practice

Google: Gemini Diffusion → DiffusionGemma

Google's path here started as a closed research demo called Gemini Diffusion, shown at Google I/O in May 2025 and limited to a waitlist. In June 2026, that research became a real, downloadable model: DiffusionGemma.

DiffusionGemma is built on the Gemma 4 family and released under an Apache 2.0 license — fully open weights. Notable details:

  • 26B total parameters as a Mixture-of-Experts model, but only about 3.8B active per inference step, so quantized versions fit within 18GB of VRAM on high-end consumer GPUs like the RTX 5090 or 4090.
  • Generates 256 tokens in parallel per forward pass with bidirectional attention, hitting over 1,000 tokens/second on a single H100 and 700+ on an RTX 5090 — roughly 4x faster than standard Gemma 4.
  • Self-corrects mid-generation by re-evaluating the whole block and fixing likely mistakes before finalizing output.
  • Multimodal input support (text, image, video), though output is text.

Google is unusually direct about the tradeoff: DiffusionGemma scores below standard Gemma 4 on established benchmarks like MMLU and coding evaluations, and the company explicitly recommends Gemma 4 for production workloads where quality matters more than raw speed. It's also worth noting the speed advantage is strongest for a single user on dedicated hardware — in large-scale cloud serving where many requests get batched together, the parallel-decoding advantage narrows considerably.

The Open-Source Diffusion LLM Scene

Outside the two companies above, a genuinely active open research ecosystem has formed:

  • LLaDA (and LLaDA 1.5): An 8B-parameter diffusion LLM trained entirely from scratch — not adapted from an existing autoregressive model — using a masked diffusion strategy. Researchers have shown it's competitive with similarly sized autoregressive models like LLaMA 3 8B on in-context learning, and LLaDA 1.5 adds preference-optimization tuning on top.
  • Dream 7B (and Dream-Coder 7B): Unlike LLaDA, Dream is initialized from an existing autoregressive checkpoint (Qwen2.5-7B) and then continually pretrained into a diffusion model. Dream-Coder 7B specializes in code and adaptively switches its generation strategy — sketching a rough structure first for complex algorithms, or generating straightforwardly left-to-right for simple completions.
  • NVIDIA Nemotron-Labs Diffusion: An open family (3B/8B/14B) notable for being dual-mode — the same weights can run as a conventional autoregressive model or switch into a faster diffusion decoding mode, giving developers a tunable speed/quality dial without retraining.
  • d3LLM and similar distillation frameworks: Research projects that distill existing diffusion models like LLaDA and Dream into much faster variants — reporting up to 5x speedups over autoregressive baselines and 10x over the original vanilla diffusion models — showing the field is still actively squeezing more performance out of the same underlying idea.

Watch: Where This Technology Is Headed

Strengths of Diffusion LLMs

  • Raw throughput: 1,000+ tokens/second is common across Mercury 2 and DiffusionGemma, versus roughly 70–250 tokens/second for speed-optimized autoregressive models. For high-volume workloads — agent loops, search reranking, transcript cleanup — this compounds into large real-world latency and cost savings.
  • Cost efficiency: Parallel generation makes better use of GPU compute that would otherwise sit idle waiting on memory bandwidth, translating to meaningfully lower per-token serving costs. Reported case studies cite latency drops of 80%+ and cost reductions around 90% when swapping in a diffusion model for a narrow, well-defined sub-task.
  • Bidirectional context and self-correction: Because every token can attend to every other token in the block, these models can catch and fix their own inconsistencies mid-generation rather than being locked into an earlier mistake, and they handle non-linear tasks (code infilling, constraint-satisfaction problems like Sudoku, in-line editing) more naturally than strictly left-to-right models.
  • Controllable, structured output: The ability to edit and generate tokens in any order makes it easier to reliably conform to schemas, JSON formats, or specific semantic constraints — useful for tool use and structured data extraction.
  • Edge and local deployment: Smaller active-parameter footprints (like DiffusionGemma's 3.8B active out of 26B total) let capable models run within consumer GPU VRAM limits, an appealing profile for on-device and edge applications.

Weaknesses of Diffusion LLMs

  • Quality still trails top frontier models on hard reasoning: Both Mercury 2 and DiffusionGemma are explicitly positioned against mid-tier or "speed-optimized" competitors (Claude Haiku, Gemini Flash), not against flagship frontier reasoning models. Google's own benchmarks show DiffusionGemma scoring measurably lower than standard Gemma 4 on things like MMLU, coding evaluations, and vision tasks.
  • The cloud-serving advantage shrinks: The dramatic speedups are strongest for single-user, low-concurrency scenarios on dedicated hardware. In production cloud environments serving many simultaneous requests via batching, the parallel-decoding edge narrows — sometimes disappearing — because batching already keeps autoregressive GPUs busy.
  • Immature tooling and ecosystem: Standard serving stacks like plain llama.cpp don't yet support diffusion sampling out of the box, requiring specialized runtimes. Fine-tuning support, agent frameworks, and community resources remain thinner than the deeply established autoregressive ecosystem.
  • Streaming behavior feels different: Chat interfaces that show text appearing token-by-token benefit from autoregressive models' immediate streaming. Diffusion models may need several denoising passes before any visible output appears, which can affect perceived responsiveness even when total generation time is faster.
  • Scaling laws are still being mapped: The autoregressive world spent years establishing clean scaling laws (more data and parameters reliably improve quality). Whether diffusion LLMs follow the same pattern, or need fundamentally different scaling strategies, is still an open research question.
  • Occasional long-form incoherence: Because tokens can be resolved without full context from later parts of a sequence, diffusion models can occasionally produce subtle inconsistencies in long-form generation — a known, actively worked-on limitation.

Where This Is Probably Heading: Hybrid Systems

A recurring idea across both industry and research circles is a hybrid architecture: use a fast diffusion model to generate an initial draft, then have a slower, higher-quality autoregressive model verify and refine only the parts that need it — conceptually similar to how speculative decoding already works, except the diffusion model proposes whole sequences rather than single tokens. Early estimates suggest this could capture 80–90% of the speed benefit while preserving frontier-level quality where it matters most.

Frequently Asked Questions

Is Mercury 2 available to the public?
Yes, via Inception's API (OpenAI-compatible), with AWS Bedrock integration reportedly in progress. Model weights themselves are not open — it's an API-only product.

Is DiffusionGemma actually open source?
Yes — it's released under Apache 2.0, with weights downloadable from Hugging Face, and deployable via Vertex AI, Nvidia NIM, or your own hardware, though standard llama.cpp support was still catching up shortly after release.

Which diffusion LLM should a developer try first?
For a hosted, production-ready reasoning model, Mercury 2 is the most mature commercial option. For open-weights experimentation, LLaDA and Dream 7B are the most established research models, while DiffusionGemma and Nemotron-Labs Diffusion are the newest entries with strong hardware-vendor backing (Google/Nvidia respectively).

Will diffusion replace autoregressive models entirely?
Nothing currently suggests a full replacement is imminent — every lab positions diffusion as complementary, best suited to latency-sensitive or structurally non-linear tasks, while autoregressive models remain the standard for maximum-quality, complex reasoning work. Hybrid draft-then-refine systems look like the more likely near-term direction than a wholesale switch.

Bottom Line

Diffusion LLMs are a genuinely different generation paradigm, not just a speed trick bolted onto existing models — and 2026 is the year they moved from research curiosity to shipping products, with Inception's Mercury 2, Google's DiffusionGemma, and a growing open-source lineup (LLaDA, Dream, Nemotron-Labs Diffusion) all live at once. The trade is consistent across every implementation: dramatically faster, cheaper generation in exchange for a real (if narrowing) quality gap against top-tier autoregressive models, plus a less mature surrounding ecosystem. For latency-critical, high-volume, or structurally non-linear workloads, that trade is already worth making. For frontier-level reasoning, autoregressive models still hold the line — for now.


This is a fast-moving area of AI research — benchmark numbers, pricing, and model availability described here reflect public information as of mid-2026 and may change as these companies ship updates.

No comments:

Post a Comment