We release The AI CUDA Engineer archive, a dataset consisting of approximately 30,000 CUDA kernels generated by The AI CUDA Engineer. It is released under the CC-By-4.0 license and can be accessed via HuggingFace and interactively visualized here.
Dataset Card
By Sakana AI, published under cc-by-4.0, revision 4edbe8d6d0b4.
The AI CUDA Engineer Archive: Agentic CUDA Kernel Discovery, Optimization & Composition
We release The AI CUDA Engineer archive, a dataset consisting of approximately 30,000 CUDA kernels generated by The AI CUDA Engineer. It is released under the CC-By-4.0 license and can be accessed via HuggingFace and interactively visualized here. The dataset is based on the Kernel tasks provided in KernelBench and includes a torch reference implementation, torch, NCU and Clang-tidy profiling data, multiple kernels per task, error messages and speedup scores against torch native and compile runtimes.
We envision that this dataset can enable post-training of open-source models to perform better CUDA-enabling modules. This includes offline Reinforcement Learning, preference optimization, and standard supervised fine-tuning.
Using the AI CUDA Engineer Archive
Structure
default 30,615 rows
| Split | Rows | Size |
|---|---|---|
| level_1 | 12,157 | 161.0 MB |
| level_2 | 12,938 | 181.3 MB |
| level_3 | 5,520 | 69.2 MB |
Details
- Repository
- SakanaAI/AI-CUDA-Engineer-Archive
- Publisher
- Sakana AI
- Task category
- Not stated by the source
- Tags
- code
- Size category
- Not stated by the source
- Languages
- Not stated by the source
- Revision
- 4edbe8d6d0b417e05aaf8ec7e23f78aecdc5516b
- Last updated
- 2025-02-20
Files
9 files, 67.7 MB in total.
Every file
| File | Type | Size | SHA-256 |
|---|---|---|---|
| level_1.parquet | Data | 24.7 MB | a74e399809f8 |
| level_2.parquet | Data | 28.3 MB | 17be1e247272 |
| level_3.parquet | Data | 11.8 MB | 076f6abcceb2 |
| README.md | Documentation | 2.7 KB | — |
| archive_table.png | Other | 321.2 KB | cf1d8a166527 |
| db_scan_kernels.png | Other | 2.2 MB | 1c9f66d6d04c |
| results_table.png | Other | 388.4 KB | 13f04d830d46 |
| .DS_Store | Repository | 6.1 KB | — |
| .gitattributes | Repository | 2.5 KB | — |
License and Download
- License
- cc-by-4.0
- Access
- No access gate
Released by Sakana AI through its official repository on Hugging Face. Read the license.