AI-CUDA-Engineer-Archive データセットは、AI によって生成された CUDA カーネルのコレクションであり、オープンソース モデルのその後のトレーニングと、より優れた CUDA 関数モジュールの開発を容易にすることを目的としています。このデータセットには 30,000 を超える CUDA カーネルが含まれており、それらはすべて AI 主導の CUDA エンジニアによって生成され、そのうち 17,000 を超えるカーネルが正しいことが検証されており、約 50% のカーネルが PyTorch のネイティブ ランタイムよりも優れています。

これには、PyTorch リファレンス実装、PyTorch/NCU/Clang-tidy パフォーマンス分析データ、各タスクの複数のカーネル バージョン、エラー情報、PyTorch ネイティブおよびコンパイルされたランタイムとのパフォーマンス比較データが含まれています。