AIAny
AI Deploy2026
Icon for item

DFlash: Block Diffusion for Flash Speculative Decoding

Enables parallel speculative decoding by using a lightweight block-diffusion draft model to produce multi-token drafts for faster, high-quality generation. Integrates with vLLM, SGLang and Transformers backends and ships draft models on Hugging Face.

Introduction

Most production LLM serving pipelines trade off either latency or sample quality when scaling throughput. DFlash offers a different lever: instead of token-by-token speculative sampling, it drafts multi-token blocks via a compact block-diffusion model and lets the full target model verify or correct those blocks in parallel — unlocking higher parallelism without a proportionate drop in output quality.

What Sets It Apart
  • Block-diffusion draft strategy: drafts contiguous token blocks (not single tokens), which increases parallel draft length and reduces synchronization overhead during speculative decoding. This means fewer confirmation rounds and better batched utilization for servers.
  • Lightweight, reusable draft models: DFlash provides pre-trained draft variants for many popular models (Qwen, LLaMA-3.1, gpt-oss, etc.), making integration faster. In practice this lowers CPU/GPU overhead compared to training large bespoke draft models from scratch.
  • Multi-backend support and real-world integrations: native examples and benchmarks for vLLM, SGLang, and Transformers backends plus guidance for attention backends (flash_attn, TRT/FA4 styles). That broad compatibility simplifies adoption in diverse serving stacks.
Who It's For & Trade-offs

Great fit if you run LLM inference at scale and want to boost throughput without fully sacrificing generation quality — especially for batched servers or low-amortized-cost deployments that can accept modest engineering integration. Also suited for teams that can host a small draft model alongside the target model (Hugging Face draft checkpoints are provided).

Look elsewhere if you cannot run an auxiliary draft model due to strict memory/instance limits, if your workload is extremely latency-sensitive at the single-request level (where extra verification steps could add jitter), or if you require a turnkey SaaS solution rather than an open-source integration.

Where It Fits

DFlash sits between model architecture research and serving optimizations: conceptually closer to speculative decoding libraries and serving infra (vLLM / SGLang), but it requires a small model-training step or using available draft checkpoints. Compared to classic one-token speculative approaches, it targets better parallelism and throughput at similar or slightly better quality under typical benchmarks.

Information

  • Websitegithub.com
  • AuthorsJian Chen, Yesheng Liang, Zhijian Liu, z-lab
  • Published date2026/01/04

More Items

GitHub
AI Infra2025

Defines a vendor-neutral JSON/YAML semantic model specification and tooling to exchange metrics, dimensions, lineage and other business semantics across analytics, AI and BI platforms; includes a core spec, validators, converters (dbt, GoodData, Salesforce) and example models.

GitHub
AI Deploy2018

Serves machine learning and deep learning models for cloud, data center, edge and embedded environments. Supports multiple frameworks and backends, dynamic and sequence batching, HTTP/gRPC APIs, Docker deployment and NVIDIA-optimized runtimes.

GitHub
AI Train2025

An asynchronous, high-throughput framework for large-scale reinforcement learning and agentic training that scales to 1T+ MoE models and 1000+ GPUs, with native verifiers integration, end-to-end SFT/RL/evals, and Slurm/Kubernetes deployment; requires NVIDIA GPUs.