ICML 2026poster0 citations

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

Shenggui Li, Chao Wang, YIKAI ZHU, Yubo Wang, Fan Yin, Shuai Shi, YefeiChen, Xiaomin Dong

Abstract

Speculative decoding mitigates the memory-bound nature of LLM decoding by using a lightweight draft model to propose multiple tokens for parallel 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 and efficient framework for training speculative decoding models with full support for EAGLE-3. SpecForge incorporates target–draft decoupling, hybrid parallelism, optimized training kernels, and tight integration with production-grade inference engines, enabling up to 9.9x faster EAGLE-3 training for Qwen3-235B-A22B compared to the baseline. We further release SpecBundle, a suite of production-grade EAGLE-3 draft models trained with SpecForge for mainstream open-source LLMs, achieving up to 4.48x end-to-end inference speedup on SGLang and addressing the scarcity of high-quality drafts. Finally, we distill a systematic study of speculative decoding training into practical and actionable recipes to guide real-world adoption.

LLMTheory
BibTeX
@inproceedings{
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 Yineng Zhang and Lei Yu and Yonggang Wen and Ivor Tsang and Tianwei Zhang},
booktitle={Forty-third International Conference on Machine Learning},
year={2026},
url={https://openreview.net/forum?id=CQOEbxy0tE}
}