Skip to content
Trending storyDeveloping

ReplaySSM caches SSM inputs instead of state to speed hybrid-model decoding

2 articles2 sourcessince Oct 12Last article 1h ago ·

Overview

AISummary of 2 articles

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

  1. Oct 12, 1:01 PM ET · 1 article
    Tri Dao explains why SSMs now gain inference optimizations by caching the right state
    Tri Dao: Tri Dao explains ReplaySSM, a caching change that speeds up SSM decoding
  2. Oct 12, 12:00 PM ET · 1 article
    ReplaySSM rethinks state storage to speed hybrid-model decoding
    NVIDIA Technical Blog: ReplaySSM rethinks state storage to speed hybrid-model decoding

Article timeline

The articles in this story. Times are ET.

Oct 12
  1. Tri DaoX
    Tri Dao explains ReplaySSM, a caching change that speeds up SSM decoding

    AITri 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.

  2. NVIDIA Technical BlogOfficial
    ReplaySSM 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.