Team Ai
Datasetpublic

mzweilin/KernelBench

KernelBench A benchmark designed to evaluate the ability of LLMs to generate efficient GPU kernels for optimizing neural network performance Version [07-21-2025] This HF dataset version has been updated to v0.1 Citation @misc{ouyang2024kernelbench, title={KernelBench: Can LLMs Write GPU Kernels?}, author={Anne Ouyang and Simon Guo and Azalia Mirhoseini}, year={2024}… See the full description on the dataset page: https://huggingface.co/datasets/mzweilin/KernelBench.

sourceHugging Faceupdated 8mo agoView on Hugging Face
0likes25downloads
Dataset Card

KernelBench

A benchmark designed to evaluate the ability of LLMs to generate efficient GPU kernels for optimizing neural network performance

Version

[07-21-2025] This HF dataset version has been updated to v0.1

Citation

bibtex
@misc{ouyang2024kernelbench,
      title={KernelBench: Can LLMs Write GPU Kernels?}, 
      author={Anne Ouyang and Simon Guo and Azalia Mirhoseini},
      year={2024},
      url={https://scalingintelligence.stanford.edu/blogs/kernelbench/}, 
}