mlsanityData to 14 Oct 2026

Towards Automated Kernel Generation in the Era of LLMs

arXiv 2601.15727 ↗First posted 22 Jan 2026cs.LG cs.CLcited by 11

On the map: LLM agents and tool use → Software engineering coding agents

Abstract (arXiv)

The performance of modern AI systems is fundamentally constrained by the quality of their underlying GPU kernels, which translate high-level algorithmic semantics into low-level hardware operations. Achieving near-optimal kernels requires expert-level understanding of hardware architectures and programming models, making kernel engineering a critical but notoriously time-consuming and non-scalable process. Recent advances in large language models and LLM-based agents have opened new possibilities for automating kernel generation and optimization. LLMs are well-suited to compress expert-level kernel knowledge that is difficult to formalize, while agentic systems further enable scalable optimization by casting kernel development as an iterative, feedback-driven loop. Rapid progress has been made in this area. However, the field remains fragmented and lacks a systematic perspective for LLM-driven kernel generation. This survey addresses this gap by providing a structured overview of existing approaches, spanning LLM-based approaches and agentic optimization workflows, and systematically organizing the datasets and benchmarks that underpin learning and evaluation in this domain. Moreover, key open challenges and future research directions are further outlined, aiming to establish a comprehensive reference for the next generation of automated kernel optimization. To keep track of this field, we maintain an open-source GitHub repository at https://github.com/flagos-ai/awesome-LLM-driven-kernel-generation.

Key facts. Cited by 11 papers in the map (data through 14 Oct 2026). Placed in Software engineering coding agents, one of 16 directions in LLM agents and tool use. Builds on 45 of its 79 references that are in the map. First posted on arXiv 22 Jan 2026.

Builds on

45 of 79 references.

Closest ancestors · 7

Same direction, newest first

  1. 2025AKG kernel Agent: A Multi-Agent Framework for Cross-Platform Kernel Synthesis29 Dec
  2. 2025cuPilot: A Strategy-Coordinated Multi-agent Framework for CUDA Kernel Evolution18 Dec
  3. 2025PRAGMA: A Profiling-Reasoned Multi-Agent Framework for Automatic Kernel Optimization9 Nov
  4. 2025CudaForge: An Agent Framework with Hardware Feedback for CUDA Kernel Optimization23 Oct
  5. 2025STARK: Strategic Team of Agents for Refining Kernels19 Oct
  6. 2025Towards Robust Agentic CUDA Kernel Benchmarking, Verification, and Optimization16 Sep
  7. 2025Astra: A Multi-Agent System for GPU Kernel Performance Optimization9 Sep

Other references · 38

  1. 2017Attention Is All You Needcited by 60,361
  2. 2020Scaling Laws for Neural Language Modelscited by 6,586
  3. 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,673
  4. 2023A Survey on Large Language Model based Autonomous Agentscited by 2,242
  5. 2022LLM.int8(): 8-bit Matrix Multiplication for Transformers at Scalecited by 906
  6. 2024StarCoder 2 and The Stack v2: The Next Generationcited by 502
  7. 2025FlashInfer: Efficient and Customizable Attention Engine for LLM Inference Servingcited by 258
  8. 2025KernelBench: Can LLMs Write Efficient GPU Kernels?cited by 225
  9. 2024Liger Kernel: Efficient Triton Kernels for LLM Trainingcited by 130
  10. 2025Kevin: Multi-Turn RL for Generating CUDA Kernelscited by 78
  11. 2025TritonBench: Benchmarking Large Language Model Capabilities for Generating Triton Operatorscited by 75
  12. 2021FBGEMM: Enabling High-Performance Low-Precision Deep Learning Inferencecited by 41
26 more
  1. 2025AutoTriton: Automatic Triton Programming with Reinforcement Learning in LLMscited by 40
  2. 2025MultiKernelBench: A Multi-Platform Benchmark for Kernel Generationcited by 38
  3. 2024Can Large Language Models Write Parallel Code?cited by 36
  4. 2025CUDA-LLM: LLMs Can Write Efficient CUDA Kernelscited by 34
  5. 2025Geak: Introducing Triton Kernel AI Agent & Evaluation Benchmarkscited by 32
  6. 2026FlashInfer-Bench: Building the Virtuous Cycle for AI-driven LLM Systemscited by 25
  7. 2025TritonRL: Training LLMs to Think and Code Triton Without Cheatingcited by 24
  8. 2025GPU Kernel Scientist: An LLM-Driven Framework for Iterative Kernel Optimizationcited by 22
  9. 2025The FM Agentcited by 21
  10. 2025QiMeng-Kernel: Macro-Thinking Micro-Coding Paradigm for LLM-Based High-Performance GPU Kernel Generationcited by 19
  11. 2025TritonForge: Profiling-Guided Framework for Automated Triton Kernel Optimizationcited by 16
  12. 2025ConCuR: Conciseness Makes State-of-the-Art Kernel Generationcited by 13
  13. 2025CUDA-L2: Surpassing cuBLAS Performance for Matrix Multiplication through Reinforcement Learningcited by 13
  14. 2026AscendKernelGen: A Systematic Study of LLM-Based Kernel Generation for Neural Processing Unitscited by 12
  15. 2024Improving Parallel Program Performance with LLM Optimizers via Agent-System Interfacescited by 11
  16. 2025SwizzlePerf: Hardware-Aware LLMs for GPU Kernel Performance Optimizationcited by 11
  17. 2025From Large to Small: Transferring CUDA Optimization Expertise via Reasoning Graphcited by 10
  18. 2025Agentic Operator Generation for ML ASICscited by 7
  19. 2025NPUEval: Optimizing NPU Kernels with LLMs and Open Source Compilerscited by 6
  20. 2025QiMeng-TensorOp: Automatically Generating High-Performance Tensor Operators with Hardware Primitivescited by 5
  21. 2025PEAK: A Performance Engineering AI-Assistant for GPU Kernels Powered by Natural Language Transformationscited by 5
  22. 2025QiMeng-Attention: SOTA Attention Operator is generated by SOTA Attention Algorithmcited by 4
  23. 2026MaxCode: A Max-Reward Reinforcement Learning Framework for Automated Code Optimizationcited by 4
  24. 2025KForge: Program Synthesis for Diverse AI Hardware Acceleratorscited by 3
  25. 2025GPU Kernel Optimization Beyond Full Builds: An LLM Framework with Minimal Executable Programscited by 3
  26. 2026DiffBench Meets DiffAgent: End-to-End LLM-Driven Diffusion Acceleration Code Generationcited by 1

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 11.

11 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026KernelCraft: Benchmarking for Agentic Close-to-Metal Kernel Generation on Emerging Hardware
  2. 2026EGG: An Expert-Guided Agent Framework for Kernel Generation
  3. 2026KForge: LLM-Driven Cross-Platform Kernel Generation for AI Accelerators
  4. 2026Kernel Forge: An Agent Harness for LLM-based Generation and Optimization of CUDA Kernels
  5. 2026K-Search: LLM Kernel Generation via Co-Evolving Intrinsic World Model
  6. 2026KernelSkill: A Multi-Agent Framework for GPU Kernel Optimization
  7. 2026Improving Efficiency of GPU Kernel Optimization Agents using a Domain-Specific Language and Speed-of-Light Guidance
  8. 2025AccelOpt: A Self-Improving LLM Agentic System for AI Accelerator Kernel Optimization