AGI Hunt· blelbach·· 3 小时前AI 评分46
Jane Street 与 NVIDIA 用 cuTile 实现 Mamba-2 高性能前向核
Jane Street 用 NVIDIA cuTile 实现 Mamba-2 高性能前向核
AI 导读
Jane Street 与 NVIDIA 联合发布技术博客，介绍基于 Python 的 tile 级 DSL cuTile 如何实现 Mamba-2 状态空间模型的高性能前向传播核函数。
来源：AGI Hunt · agihunt.info