跳到正文
NVIDIA 开发者博客· Michelle Horton·· 3 小时前精选AI 评分55

ReplaySSM:重新思考混合模型解码的状态存储

ReplaySSM: Rethinking State Storage for Faster Hybrid-Model Decoding

AI 导读

NVIDIA 开发者博客介绍 ReplaySSM,围绕混合模型中状态空间模型(SSM)层与少量注意力层交替的结构,重新思考状态存储方式。

推荐理由

文章梳理了混合模型用固定大小状态替代KV缓存的设计取舍,可帮助理解解码加速的思路。

来源:NVIDIA 开发者博客 · developer.nvidia.com