DICE: Diffusion Large Language Models Excel at Generating CUDA Kernels
Haolei Bai1, Lingcheng Kong1,2, Xueyi Chen1, Jiamian Wang3, Zhiqiang Tao3, Huan Wang1,β
1Westlake University, 2The Hong Kong University of Science and Technology, 3Rochester Institute of Technology
β Corresponding author: wanghuan@westlake.edu.cn
- We will release our code soon. Stay tuned!
- 2026.02.13: We release DICE-1.7B, DICE-4B, and DICE-8B on Hugging Face !
- 2026.02.13: The paper is on arXiv !
We are grateful to the SDAR, TraceRL, KernelBench, cudaLLM for releasing their code publicly, which greatly facilitated our work.
If you find DICE useful for your research or projects, please consider citing our work:
@article{bai2026dice,
title={DICE: Diffusion Large Language Models Excel at Generating CUDA Kernels},
author={Bai, Haolei and Kong, Lingcheng and Chen, Xueyi and Wang, Jiamian and Tao, Zhiqiang and Wang, Huan},
journal={arXiv preprint arXiv:2602.11715},
year={2026}
}