SpecForge: A Flexible and Efficient Open-Source Training Framework for Speculative Decoding

Shenggui Li, Chao Wang, Yikai Zhu, Yubo Wang, Fan Yin +12 more
3/19/2026
cs.LGcs.AIcs.CL

No confident code implementation found for this paper.

Abstract

Large language models incur high inference latency due to sequential autoregressive decoding. Speculative decoding alleviates this bottleneck by using a lightweight draft model to propose multiple tokens for batched verification. However, its adoption has been limited by the lack of high-quality draft models and scalable training infrastructure. We introduce SpecForge, an open-source, production-oriented framework for training speculative decoding models with full support for EAGLE-3. SpecForge incorporates target-draft decoupling, hybrid parallelism, optimized training kernels, and integration with production-grade inference engines, enabling up to 9.9x faster EAGLE-3 training for Qwen3-235B-A22B. In addition, we release SpecBundle, a suite of production-grade EAGLE-3 draft models trained with SpecForge for mainstream open-source LLMs. Through a systematic study of speculative decoding training recipes, SpecBundle addresses the scarcity of high-quality drafts in the community, and our draft models achieve up to 4.48x end-to-end inference speedup on SGLang, establishing SpecForge as a practical foundation for real-world speculative decoding deployment.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. Any repos shown below are weak matches — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{li2026specforge,
  title  = {SpecForge: A Flexible and Efficient Open-Source Training Framework for Speculative Decoding},
  author = {Shenggui Li and Chao Wang and Yikai Zhu and Yubo Wang and Fan Yin and Shuai Shi and Yefei Chen and Xiaomin Dong and Qiaoling Chen and Jin Pan and Ji Li and Laixin Xie and Yineng Zhang and Lei Yu and Yonggang Wen and Ivor Tsang and Tianwei Zhang},
  year   = {2026},
  eprint = {2603.18567},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2603.18567v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2603.18567.svg)](https://codeofpaper.com/paper/2603.18567)

Discussion