Kernel-Smith: A Unified Recipe for Evolutionary Kernel Optimization
Not on the map. Why?
Abstract (arXiv)
We present Kernel-Smith, a framework for high-performance GPU kernel and operator generation that combines a stable evaluation-driven evolutionary agent with an evolution-oriented post-training recipe. On the agent side, Kernel-Smith maintains a population of executable candidates and iteratively improves them using an archive of top-performing and diverse programs together with structured execution feedback on compilation, correctness, and speedup. To make this search reliable, we build backend-specific evaluation services for Triton on NVIDIA GPUs and Maca on MetaX GPUs. On the training side, we convert long-horizon evolution trajectories into step-centric supervision and reinforcement learning signals by retaining correctness-preserving, high-gain revisions, so that the model is optimized as a strong local improver inside the evolutionary loop rather than as a one-shot generator. Under a unified evolutionary protocol, Kernel-Smith-235B-RL achieves state-of-the-art overall performance on KernelBench with Nvidia Triton backend, attaining the best average speedup ratio and outperforming frontier proprietary models including Gemini-3.0-pro and Claude-4.6-opus. We further validate the framework on the MetaX MACA backend, where our Kernel-Smith-MACA-30B surpasses large-scale counterparts such as DeepSeek-V3.2-think and Qwen3-235B-2507-think, highlighting potential for seamless adaptation across heterogeneous platforms. Beyond benchmark results, the same workflow produces upstream contributions to production systems including SGLang and LMDeploy, demonstrating that LLM-driven kernel optimization can transfer from controlled evaluation to practical deployment.
Key facts. Cited by 5 papers in the map (data through 13 Oct 2026). Builds on 26 of its 35 references that are in the map. First posted on arXiv 30 Mar 2026.
Builds on
26 of 35 references.
References · 26
- 2025Qwen3 Technical Reportcited by 8,510
- 2021Evaluating Large Language Models Trained on Codecited by 8,205
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,831
- 2023Code Llama: Open Foundation Models for Codecited by 2,130
- 2019Megatron-LM: Training Multi-Billion Parameter Language Models Using Model Parallelismcited by 1,851
- 2023SGLang: Efficient Execution of Structured Language Model Programscited by 1,117
- 2025AlphaEvolve: A coding agent for scientific and algorithmic discoverycited by 819
- 2025DeepSeek-V3.2: Pushing the Frontier of Open Large Language Modelscited by 784
- 2026Kimi K2.5: Visual Agentic Intelligencecited by 472
- 2015Illuminating search spaces by mapping elitescited by 415
- 2025KernelBench: Can LLMs Write Efficient GPU Kernels?cited by 213
- 2026Learning to Discover at Test Timecited by 104
14 more
- 2026Conditional Memory via Scalable Lookup: A New Axis of Sparsity for Large Language Modelscited by 81
- 2025Kevin: Multi-Turn RL for Generating CUDA Kernelscited by 76
- 2025CUDA-L1: Improving CUDA Optimization via Contrastive Reinforcement Learningcited by 61
- 2025Astra: A Multi-Agent System for GPU Kernel Performance Optimizationcited by 49
- 2025CudaForge: An Agent Framework with Hardware Feedback for CUDA Kernel Optimizationcited by 45
- 2026CUDA Agent: Large-Scale Agentic RL for High-Performance CUDA Kernel Generationcited by 43
- 2025AutoTriton: Automatic Triton Programming with Reinforcement Learning in LLMscited by 39
- 2025MultiKernelBench: A Multi-Platform Benchmark for Kernel Generationcited by 37
- 2026Dr. Kernel: Reinforcement Learning Done Right for Triton Kernel Generationscited by 21
- 2026K-Search: LLM Kernel Generation via Co-Evolving Intrinsic World Modelcited by 21
- 2025PRAGMA: A Profiling-Reasoned Multi-Agent Framework for Automatic Kernel Optimizationcited by 11
- 2026KernelSkill: A Multi-Agent Framework for GPU Kernel Optimizationcited by 11
- 2026CUDABench: Benchmarking LLMs for Text-to-CUDA Generationcited by 10
- 2026Fine-Tuning GPT-5 for GPU Kernel Generationcited by 4
Built on by
Citing papers per year: 2026: 5.
5 papers.
- 2026KernelBench-Verified: Do LLM-Generated Kernels Actually Beat PyTorch?cited by 8
- 2026Evolution Fine-Tuning: Learning to Discover Across 371 Optimization Taskscited by 4
- 2026daVinci-kernel: Co-Evolving Skill Selection, Summarization, and Utilization via RL for GPU Kernel Optimizationcited by 3
- 2026PTXBench: Benchmarking and Adapting LLMs for GPU Kernel Optimization with Architecture-specific PTXcited by 0
- 2026CUDA-Harness: Harnessing Agentic CUDA Kernel Generation and Optimization from Natural Languagecited by 0
- 2026CUDA-Harness: Harnessing Agentic CUDA Kernel Generation and Optimization from Natural Language30 Aug 2026
- 2026PTXBench: Benchmarking and Adapting LLMs for GPU Kernel Optimization with Architecture-specific PTX18 Aug 2026
- 2026Evolution Fine-Tuning: Learning to Discover Across 371 Optimization Tasks27 Jun 2026
- 2026KernelBench-Verified: Do LLM-Generated Kernels Actually Beat PyTorch?26 Jun 2026
- 2026daVinci-kernel: Co-Evolving Skill Selection, Summarization, and Utilization via RL for GPU Kernel Optimization15 Jun 2026