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.cuda-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.cudakernelbench-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.Nemotron-SFT-CUDA-v1
Dataset Description:
Nemotron-SFT-CUDA-v1 is a training dataset for CUDA code. It helps language models write CUDA kernels and solve CUDA programming problems.
We start from CUDA code in Nemotron Pretraining Code v2, which has a permissive license. An OpenCode agent powered by GLM-4.7 reads that code and writes new CUDA programming problems. Each problem comes with a hidden answer and tests. A second OpenCode + GLM-4.7 agent then tries to solve the problems using only the prompt… See the full description on the dataset page: https://huggingface.co/datasets/nvidia/Nemotron-SFT-CUDA-v1.