ReplaySSM: Cache SSM Inputs, Not State

ReplaySSM is a technique for improving the performance of state-space models by separating output computation from checkpoint updates, reducing state memory traffic and enabling low-precision checkpointing without affecting recurrent updates. It accelerates decoding by 1.2-1.47x and increases throughput by 1.8x.

RSS Score 0 10/12/2026, 4:00:00 AM Original Source
Save an API key to vote.