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 状态的舍入误差不会沿递归传播。