突破 Blackwell 兼容性瓶颈：SM1 实现纯 PyTorch 版 Mamba 架构

开发者成功构建了名为 SM1（Scalar Mamba1）的变体，通过数学闭式解将 Mamba 的核心选择性扫描（Selective Scan）简化为原生 PyTorch 算子，解决了该架构在 NVIDIA Blackwell (sm_120) 硬件及 Windows 环境下的编译难题。

• ▶ 硬件解耦：SM1 彻底摆脱了对特定 CUDA 内核（mamba-ssm）的依赖，利用原生 cumprod 和 cumsum 算子实现了与原始算法数学一致的逻辑。
• ▶ 架构简化：通过常数变易法（Method of Variation of Parameters）推导出 d_state=1 递归的精确解，证明了在特定维度下，复杂的状态空间模型（SSM）可以被极简实现。

八卦洞察

SM1 的出现揭示了当前 AI 基础设施的一个痛点：前沿架构（如 SSM）往往过度依赖高度优化的定制化 CUDA Kernel，这导致了严重的硬件滞后性——即便是最先进的 Blackwell 显卡，在初期也面临驱动和算子库不匹配的尴尬。SM1 放弃了高维状态（d_state > 1）带来的微弱表达力增益，换取了在 Blackwell 上的“即插即用”能力。这种“以退为进”的工程思路，对于需要在非 Linux 环境或最新硬件上快速部署 Mamba 模型的团队具有极高的参考价值。

行动建议

• 工程团队：若在 Windows 或新一代 NVIDIA 硬件上遇到 mamba-ssm 编译失败，应优先评估 SM1 这种纯 PyTorch 实现方案，以降低环境配置成本。
• 研究人员：关注 d_state=1 在大规模任务中的表现损耗。如果标量状态足以支撑特定领域的性能，那么 SSM 的计算复杂度将进一步下降，有利于边缘侧推理。