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