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

Measures generative AI inference performance with token-level metrics (TTFT, inter-token latency), latency, and throughput under realistic traffic patterns. Provides a multiprocess engine, real-time TUI dashboard, extensible plugins, and integrations for telemetry and result uploads, aimed at inference benchmarking and capacity planning.

GitHub
AI Train2019

Train and experiment with multi-billion to trillion-parameter transformer models on large GPU clusters using GPU-optimized building blocks and reference training scripts; offers advanced parallelism and mixed-precision support for research teams and ML engineers.

GitHub

Indexes full text of visited web pages and local files on a self‑hosted server so you can search your personal knowledge from a web UI, terminal, CLI, or an AI assistant. Runs without mandatory telemetry, offers a browser extension for automatic capture, and supports optional semantic search via a configurable embeddings endpoint.