CUDA
Datasets
All datasets matching “CUDA”AI-CUDA-Engineer-Archive
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… See the full description on the dataset page: https://huggingface.co/datasets/SakanaAI/AI-CUDA-Engineer-Archive.cudacuda-runtime-bundle
MBFS CUDA Runtime Bundle
Self-contained, ABI-matched CUDA execution stacks for running
ONNX Runtime on Windows with the CUDA / TensorRT
execution providers. One folder per CUDA major version:
Stack
Folder
CUDA
cuDNN
ONNX Runtime EP
When to use
cu12
cuda_v12/
12.x
9
CUDA + TensorRT 10
sm_75+ GPU on driver R527+ (CUDA 12)
cu11
cuda_v11/
11.x
8
CUDA + TensorRT 8.6
Pascal/Volta (sm_60–sm_70), or any GPU on a driver capped at CUDA 11.x
Pick the stack matching your… See the full description on the dataset page: https://huggingface.co/datasets/MBFSAITeam/cuda-runtime-bundle.kernelbench-cuda-tracesdim58-cudaData-31cases
Dim58 CPU Data — 31 Cases
Dataset uploaded from:
/mnt/data/ubuntu/research/outputs/data_cuda_geodesic58
Dataset summary
Property
Value
Repository
hosseinbv/dim58-cudaData-31cases
Number of files
64
Total size
17.99 GB
Source folder
data_cuda_geodesic58
File types
Extension
File count
.npz
62
.json
1
.csv
1
Top-level contents
0000_internal_case1_data.npz
0001_internal_B_10.npz… See the full description on the dataset page: https://huggingface.co/datasets/hosseinbv/dim58-cudaData-31cases.CUDA-Agent-Ops-6K
CUDA-Agent-Ops-6K
CUDA-Agent-Ops-6K is a curated training dataset for CUDA kernel generation and optimization.
It is released as part of the CUDA-Agent project:
Project Page: https://CUDA-Agent.github.io/
Github Repo: https://github.com/BytedTsinghua-SIA/CUDA-Agent
Dataset Summary
CUDA-Agent-Ops-6K contains 6,000 synthesized operator-level training tasks designed for large-scale agentic RL training. It is intended to provide diverse and executable CUDA-oriented training… See the full description on the dataset page: https://huggingface.co/datasets/BytedTsinghua-SIA/CUDA-Agent-Ops-6K.