768 次浏览 · 3 赞 · 0 转发 (收录时)
ReplaySSM 报告解码加速和生产应用,vLLM 已提供支持
发生了什么
vLLM 称赞了 @zwljohnny、@tri_dao 和 @NVIDIA 的 ReplaySSM 工作,并表示 Mamba2 标准解码和基于 FlashInfer 的 MTP 支持已在 vLLM 中提供。 缓存 SSM 输入,并仅在需要时更新检查点,可减少循环状态的内存流量。 在被引用的帖子中,@zwljohnny 报告称,ReplaySSM 已用于 GLM-5.3-Flash 和 Kimi-K3 的生产环境,并已在 TensorRT-LLM、vLLM、FlashInfer 和 SGLang 中提供;标准解码最高提速 1.47x,推测解码最高提速 2.1x;低精度 SSM 状态的舍入误差不会沿递归传播。
这件事到目前为止
vLLM 表示,vLLM 已支持 ReplaySSM,可用于 Mamba2 标准解码和由 FlashInfer 支持的多 token 预测(MTP)。它通过缓存状态空间模型(SSM)的输入,并仅在需要时更新检查点,减少循环状态的内存流量。 在 vLLM 引述的帖子中,@zwljohnny 报告称,ReplaySSM 已用于 GLM-5.3-Flash 和 Kimi-K3 的生产环境,并已在 TensorRT-LLM、vLLM、FlashInfer 和 SGLang 中提供。@zwljohnny 报告称,标准解码速度最高提升 1.47x,推测解码速度最高提升 2.1x;还表示 ReplaySSM 支持低精度 SSM 状态,其舍入误差不会在递推过程中传播。
继续阅读
最新日报 · 第 24 期NVIDIA 与 AI4Bharat 推出 Nemotron-pretraining-Indic-v1 及开放数据生成流水线正在热议
明早 08:00 把这样一份日报发到你邮箱
和上面这份一样:几件值得看的事。免费,点确认信之后才会开始发送,每封信都能一键退订。 看一份往期日报 · 2026-10-17
