Skip to content

DICE: Diffusion Large Language Models Excel at Generating CUDA Kernels

Notifications You must be signed in to change notification settings

deadlykitten4/DICE

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

10 Commits
Β 
Β 
Β 
Β 

Repository files navigation

DICE: Diffusion Large Language Models Excel at Generating CUDA Kernels

arXiv Hugging Face GitHub Repo stars


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

πŸ—“οΈ Plan

  • 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 !

πŸ™Œ Acknowledgement

We are grateful to the SDAR, TraceRL, KernelBench, cudaLLM for releasing their code publicly, which greatly facilitated our work.

πŸ“– Citation

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}
}

About

DICE: Diffusion Large Language Models Excel at Generating CUDA Kernels

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published