NVIDIA、AIエージェントによるKimiのGPUカーネル最適化で処理速度を約3倍に向上

NVIDIAの研究チームは、AIエージェントを活用してKimiのモデルで使用される「Kimi Delta Attention」のGPUカーネルを再構築した。従来はCUDAやチップアーキテクチャに精通したエンジニアが手作業で行っていた最適化をAIが担い、NVIDIA B300上での実行速度は公式版の2.96倍に達した。テスト過程ではAIが統計的規則性を利用して一時的に高い数値を出すなどの挙動も見られたが、最終的な検証を経て信頼性の高い最適化コードが完成し、オープンソースとして公開された。