Search papers, labs, and topics across Lattice.
The daVinci-kernel framework introduces a novel reinforcement learning approach for GPU kernel optimization by dynamically evolving a skill library that integrates skill selection, summarization, and utilization. It employs three specialized agents鈥擲kill Selection, Policy, and Skill Summary鈥攕haring a common LLM backbone to enhance the efficiency of CUDA/Triton kernel generation through multi-turn interactions and execution-based skill verification. This method achieved significant performance improvements on KernelBench, surpassing the previous state-of-the-art model, Dr.Kernel-14B, with notable gains across multiple optimization levels.
daVinci-kernel outperforms the best existing RL-trained model in GPU kernel optimization by effectively co-evolving skill selection and execution strategies.
GPU kernel optimization represents a paradigm where functional correctness is assumed and execution efficiency is the objective. We present daVinci-kernel, a reinforcement learning framework that couples skill discovery with skill exploitation through a dynamically evolving skill library. daVinci-kernel jointly trains three agents sharing one LLM backbone: a Skill Selection Agent that retrieves relevant techniques via BM25 and LLM reranking, a Policy Agent that generates multi-turn CUDA/Triton kernels conditioned on selected skills, and a Skill Summary Agent that distills successful rollouts into reusable skills. Candidate skills are added only after execution-based verification confirms reproducible speedups. All three agents share a single LLM backbone, are initialized via a structured SFT cold start on diversity-filtered data, and are then jointly optimized end-to-end with multi-turn REINFORCE and per-agent advantage estimation. On KernelBench, daVinci-kernel-14B achieves 37.2%, 70.6%, and 32.2% on Level 1, Level 2, and Level 3 under the Fast$_1$ threshold, outperforming the strongest prior RL-trained model, Dr.Kernel-14B.
