AI-CUDA-Engineer-Archive 数据集是一个由 AI 生成的 CUDA 内核集合，旨在促进开源模型的后续训练和开发更出色的 CUDA 功能模块。该数据集包含超过 3 万个 CUDA 内核，这些内核均由 AI 驱动的 CUDA 工程师生成，其中超过 1.7 万个内核已验证正确，约 50% 的内核性能优于 PyTorch 原生运行时间。 它包括 PyTorch 参考实现、 PyTorch/NCU/Clang-tidy 性能分析数据、每个任务的多个内核版本、错误信息以及与 PyTorch 原生和编译运行时间的性能对比数据。

Citation

@article{lange2025aicudaengineer, title={The AI CUDA Engineer: Agentic CUDA Kernel Discovery, Optimization and Composition}, author={Lange, Robert Tjarko and Prasad, Aaditya and Sun, Qi and Faldor, Maxence and Tang, Yujin and Ha, David}, journal={arXiv preprint}, year={2025} }