datasets
Training and evaluation data, with the modality, task and licence stated up front. Listed live from the Hugging Face Hub.
triton-gpu-latency
Triton GPU Latency Dataset
A large dataset of PyTorch problems (mostly from KernelBench) paired with candidate Triton-kernel implementations and their measured GPU runtimes generated by MakoraGenerate. Each row is a self-contained Python program that defines (1) a reference Model written with plain PyTorch ops and (2) a ModelNew that re-implements the same forward pass with a hand-written or generated Triton kernel. The label is the runtime of executing ModelNew.
Built for… See the full description on the dataset page: https://huggingface.co/datasets/makora-ai/triton-gpu-latency.cuda-triton-gpu-kernels-2026
⚡ Complete 2026 CUDA & OpenAI Triton High-Performance GPU Kernel Engineering SFT/DPO Suite
The definitive, production-grade synthetic alignment dataset engineered for training and fine-tuning open-weights Large Language Models (Qwen 2.5 Coder, DeepSeek-Coder, Llama 3.1) on ultra-high-throughput GPU kernel programming: NVIDIA Hopper H100 / Blackwell B200 TMA async transfers, OpenAI Triton 3.1+ FlashAttention-3, 32-bank conflict elimination, and low-bit FP8 / INT4 GEMM… See the full description on the dataset page: https://huggingface.co/datasets/beatsprom/cuda-triton-gpu-kernels-2026.Code_Opt_Triton
Overview
This dataset, TEEN-D/Code_Opt_Triton, is an extended version of the publicly available GPUMODE/Inductor_Created_Data_Permissive dataset. It contains pairs of original (PyTorch or Triton) programs and their equivalent Triton code (generated by torch inductor), intended for training models in PyTorch-to-Triton code translation and optimization.
The primary modification in this extended version is that each optimized Triton code snippet is paired with both its original source… See the full description on the dataset page: https://huggingface.co/datasets/Teen-Different/Code_Opt_Triton.qwen38-27b-triton-ascend-rl-trajectories
Qwen3.8-27B Triton-Ascend RL Trajectories
This dataset contains 1,000 multi-turn trajectories for Triton-Ascend kernel generation. Every included trajectory passed compilation and correctness validation on one official npu-kernelbench workload. Qwen3.8-27B generated an initial solution and received evaluator feedback for up to five calls.
Dataset Viewer subsets
trajectories (default): one row per sample with only messages. The initial system and task user… See the full description on the dataset page: https://huggingface.co/datasets/Yukki1011/qwen38-27b-triton-ascend-rl-trajectories.kernelbook-triton-multiturn-reasoning-traces
KernelBench Triton Multi-Turn Reasoning Traces
A dataset of multi-turn reasoning traces for Triton GPU kernel generation from PyTorch reference implementations. Each trace captures the full iterative refinement loop — model reasoning, generated kernel code, execution feedback, and benchmark results.
Generation Setup
Model & Serving
Problems were sent to Qwen3-235B-A22B-Thinking-2507 (FP8) served via vLLM on H100 GPUs (tensor parallel, 131k context window). Reasoning… See the full description on the dataset page: https://huggingface.co/datasets/ppbhatt500/kernelbook-triton-multiturn-reasoning-traces.kernelbook-triton-reasoning-traces
KernelBench Triton Reasoning Traces
Reasoning traces generated by the gpt-oss-120b model for converting PyTorch modules to Triton GPU kernels.
Dataset Description
This dataset contains 170 reasoning traces around 85% of them are correct where a PyTorch module was successfully converted to a Triton kernel. Each sample includes the original PyTorch code, the model's reasoning process, and the resulting Triton kernel code along with correctness and performance benchmarks.… See the full description on the dataset page: https://huggingface.co/datasets/ppbhatt500/kernelbook-triton-reasoning-traces.autonomous-gpu-kernel-triton-cuda-suite-2026
⚡ Autonomous GPU Kernel, Triton & CUDA Architecture Suite (2026)
A Production-Grade, Verifiable Synthetic Corpus for Training Frontier Coding Models (Qwen 3.8, DeepSeek-V3, Llama 3.3)
⚡ Overview & Industry Problem
Modern deep learning accelerators, custom ASICs, and high-performance computing clusters demand specialized, autonomous GPU kernel infrastructure: OpenAI Triton fused kernels, FlashAttention-3 forward/backward online softmax… See the full description on the dataset page: https://huggingface.co/datasets/beatsprom/autonomous-gpu-kernel-triton-cuda-suite-2026.triton-code-dataset
Triton / CUDA Kernel Code Dataset (FIM + CUDA→Triton)
License and Provenance
Parts of this dataset include source code originally licensed under the
BSD-3-Clause License by NVIDIA Corporation and other contributors.
All license headers and copyright notices are preserved as required.
This dataset provides supervised fine-tuning splits for GPU kernel LLMs.
Files
fim_sft.csv — FIM pairs
cu2triton_sft.csv — CUDA→Triton pairs
ANALYSIS.json — stats summary
figs/ —… See the full description on the dataset page: https://huggingface.co/datasets/edwixx/triton-code-dataset.triton-sft-dataset
Triton Kernel SFT Dataset
This dataset contains 5958 examples for supervised fine-tuning (SFT) of models to generate Triton GPU kernels.
Dataset Description
The dataset consists of two types of examples:
Synthetic queries (60%): Generated queries asking for Triton kernels for various operations
Convert queries (40%): PyTorch code conversion requests to Triton kernels
Dataset Structure
Each example contains:
prompt: The instruction/query asking for a Triton… See the full description on the dataset page: https://huggingface.co/datasets/cdreetz/triton-sft-dataset.triton-sft-dataset-6k-v2
Triton Kernel SFT Dataset
This dataset contains 5995 examples for supervised fine-tuning (SFT) of models to generate Triton GPU kernels.
Dataset Description
The dataset consists of two types of examples:
Synthetic queries: 3595 examples (60.0%) - Generated queries asking for Triton kernels for various operations
Convert queries: 2400 examples (40.0%) - PyTorch code conversion requests to Triton kernels
Documentation Usage
With Triton docs: 3000 examples… See the full description on the dataset page: https://huggingface.co/datasets/cdreetz/triton-sft-dataset-6k-v2.