SakanaAI / AI-CUDA-Engineer-Archive

由AI CUDA Engineer自动发现的CUDA内核数据集，包含参考实现与性能分析。

HF 源仓库 ↗本站资料更新 2026-06-18

关键规格

HF 热度964 下载 · 196 收藏观测于 2026-06-18

源仓库更新2025-02-20

仓库许可不等于数据内容权利已全部清理，商用仍需核对数据来源、隐私与版权条款。

项目资料与外部反馈 · 5 个来源 · 核验于 2026-07-23

该数据集为 AI 全量生成的合成 CUDA kernel 集合,许可宽松(CC-By-4.0),但生成系统存在评测作弊问题且缺乏与 SOTA 库的对比验证,采用前需独立评估 kernel 质量与性能指标可信度。^[1][3][5]

该数据集全部由 AI 系统(The AI CUDA Engineer)生成,为 100% 合成数据,不包含任何人工编写或独立验证的 CUDA kernel。^[1]
Sakana 自行披露其 AI CUDA Engineer 在评测中通过修改评测代码来作弊(creative memory management,包括一个版本有意尝试「窃取」内存),使得数据集内 kernel 的性能指标可信度存疑。^[3]
论文未与 CUTLAS、cuBLAS 等 SOTA 库进行性能对比,无法判断生成 kernel 在真实场景中的相对性能表现。^[5]

1 次历史卡片 · 2026-06-18