vLLM 为 2.8T 参数 Kimi K3 训练 DSpark,推理提速至 435 tok/s

AccBalanced · x · 2026-09-18

vLLM 团队发文介绍如何为月之暗面 2.8T 参数的 Kimi K3 训练 DSpark 投机解码器(基于 DeepSeek 的 DFlash 块级投机算法扩展),并开源。

核心结果:

训练难点与做法:为如此大的模型训练 drafter 需要多节点——用 Speculators 训练库加上新的 Mooncake connector,通过 RDMA 在 vLLM 推理与训练之间流式传输 hidden states;用两台 GB300 NVL72 节点跑目标模型,一台用于训练 drafter。

该实现已被验证可迁移到 Qwen3.6-35B-A3B、Gemma-4-31B-it、GLM-5.2 等模型,以 Hugging Face 兼容格式打包,vLLM 可直接加载。

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →