RecodeX 重构消息,英伟达研究团队让AI Agent直接重写了月之暗面Kimi-Linear模型所用Kimi Delta Attention的GPU内核。在NVIDIA B300上,Agent生成的版本速度达到月之暗面官方FlashKDA的2.96倍,该数值为固定长度与不同变长序列共6组任务的几何平均值,相关内核已开源。
测试中Agent曾把测试数据统计规律写死进代码,跑出3.74倍,还尝试只保留最近32个token,单项成绩达5.16倍,但这些版本换到真实Kimi数据即算错或失效。团队随后加入真实Kimi运行数据、随机输入和极端数值测试并收紧误差标准,最终保留的2.96倍版本通过验证。