ReplaySSM caches SSM inputs instead of state to speed hybrid-model decoding
Overview
NVIDIA's technical blog presents ReplaySSM, a method that changes how state-space model layers store state to speed decoding in hybrid models.
The post names NVIDIA Nemotron 3, Qwen3.5 through Qwen3.8, Kimi-K3, and GLM-5.3-Flash as hybrids that interleave SSM layers with a few attention layers, and covers SSM layers based on Mamba-2, Gated DeltaNet, and Kimi Delta Attention. The source excerpt does not include benchmark results.
Tri Dao says ReplaySSM caches SSM inputs instead of the state, which separates output computation from state updates during decoding. He says the method runs in production in GLM-5.3-Flash and Kimi-K3 and ships in TensorRT-LLM, vLLM, FlashInfer, and SGLang. In a quoted post, he reports decoding up to 1.47x faster and speculative decoding up to 2.1x faster; these are his figures, not independently verified.
Written by AI from the articles below · updated Oct 12, 1:16 PM ET
Check the sources:
Developments
2 developments
- Oct 12, 1:01 PM ET · 1 articleTri Dao explains why SSMs now gain inference optimizations by caching the right stateTri Dao: Tri Dao explains ReplaySSM, a caching change that speeds up SSM decoding
- Oct 12, 12:00 PM ET · 1 articleReplaySSM rethinks state storage to speed hybrid-model decodingNVIDIA Technical Blog: ReplaySSM rethinks state storage to speed hybrid-model decoding
Article timeline
The articles in this story. Times are ET.
Tri Dao@tri_daoXTri Dao explains ReplaySSM, a caching change that speeds up SSM decodingAITri Dao says SSMs had good asymptotics but lacked inference tools like speculative decoding and quantization that Transformers get. He credits caching SSM inputs instead of the state, which separates output computation from state updates during decoding. The quoted post says ReplaySSM runs in production in GLM-5.3-Flash and Kimi-K3 and ships in TensorRT-LLM, vLLM, FlashInfer, and SGLang, with decoding up to 1.47x faster and speculative decoding up to 2.1x faster.
- NVIDIA Technical BlogOfficialReplaySSM rethinks state storage to speed hybrid-model decoding
AINVIDIA's technical blog presents ReplaySSM, a method that changes how state-space model layers store state to speed decoding in hybrid models. The post says hybrid models such as NVIDIA Nemotron 3, Qwen3.5 through Qwen3.8, Kimi-K3, and GLM-5.3-Flash interleave SSM layers with a few attention layers, using Mamba-2, Gated DeltaNet, or Kimi Delta Attention as the SSM layer. The source excerpt provided does not give benchmark results, so the speed gains are not quantified here.
Heat trend
Not enough continuous observations to show a trend yet.