Inference Deep dive
ReplaySSM speeds hybrid-model decoding with checkpointed state and input buffers
ReplaySSM stores a checkpoint plus recent inputs, cutting state-memory traffic roughly in half and speeding decoding kernels 1.46–1.80x; speculative throughput rises 1.85–2.10x. The design targets memory traffic, rollback, and precision trade-offs in hybrid-model serving.
- WhatReplaySSM represents SSM state as a checkpoint plus a buffer of recent inputs.
- HowIt computes outputs from the checkpoint and buffer, writing a new checkpoint when the buffer fills.
- Why it mattersSeparating output computation from state updates targets state traffic, speculative-decoding rollback, and quantization error accumulation.