Researchers have introduced KernelMosaic, an open dataset and agentic framework designed to improve the generation of CUDA kernels by large language models. The dataset pairs diverse CUDA tasks with verified solutions and optimization trajectories, curated through human review and diversity-aware selection. An agentic framework guides LLMs through implementation, repair, and performance optimization using reusable CUDA guidance and execution feedback. Fine-tuning a small open-source model on KernelMosaic significantly boosted correctness on the KernelBench benchmark and increased the fraction of tasks with kernels faster than PyTorch eager, demonstrating the dataset's effectiveness in enhancing CUDA kernel generation capabilities. AI
IMPACT Enhances LLM capabilities in generating specialized code, potentially accelerating development in GPU-intensive computing.
RANK_REASON The cluster describes a new academic paper introducing a dataset and framework for AI model training. [lever_c_demoted from research: ic=1 ai=1.0]
- alphaXiv
- arXiv
- CatalyzeX
- CUDA
- DagsHub
- Gotit.pub
- Hugging Face
- IArxiv
- KernelBench
- KernelMosaic
- PyTorch
- ScienceCast
AI-generated summary · Google Gemini · from 1 sources. How we write summaries →