Live draft model training for speculative decoding
Jun 25, 2026
Summary
Introduced a distributed training pipeline for live draft model training that extracts hidden states directly from live inference to train draft models on the fly. This pipeline addresses storage overhead, compute bottlenecks, and alignment drift associated with traditional draft model training. Key technical implementations include using mmap-backed arrays for buffering training data, offloading network communication and data buffering to a background process with CUDA event synchronization, and utilizing UCXX for RDMA transfers and Trio for structured concurrency and error handling in a distributed cluster.
More from the Baseten AI Model Deployment and Serving Platform trail
- LangChain trains custom models for LangSmith Engine with Baseten Loops
- DeepSeek-V4.1-Flash: more efficient prefill for coding agents
- Blaxel is joining Baseten to build the future of agentic infrastructure
- How Baseten makes pyannote’s diarization models 9.6x faster
- Optimizing delta weight syncs for managed rollouts