CUDA Agent: Large-Scale Agentic RL for High-Performance CUDA Kernel Generation
On the map: LLM agents and tool use → Software engineering coding agents
Abstract (arXiv)
GPU kernel optimization is fundamental to modern deep learning but remains a highly specialized task requiring deep hardware expertise. Despite strong performance in general programming, large language models (LLMs) remain uncompetitive with compiler-based systems such as torch.compile for CUDA kernel generation. Existing CUDA code generation approaches either rely on training-free refinement or fine-tune models within fixed multi-turn execution-feedback loops, but both paradigms fail to fundamentally improve the model's intrinsic CUDA optimization ability, resulting in limited performance gains. We present CUDA Agent, a large-scale agentic reinforcement learning system that develops CUDA kernel expertise through three components: a scalable data synthesis pipeline, a skill-augmented CUDA development environment with automated verification and profiling to provide reliable reward signals, and reinforcement learning algorithmic techniques enabling stable training. CUDA Agent achieves state-of-the-art results on KernelBench, delivering 100%, 100%, and 92% faster rate over torch.compile on KernelBench Level-1, Level-2, and Level-3 splits, outperforming the strongest proprietary models such as Claude Opus 4.5 and Gemini 3 Pro by about 40% on the hardest Level-3 setting.
Key facts. Cited by 41 papers in the map (data through 7 Oct 2026). Placed in Software engineering coding agents, one of 16 directions in LLM agents and tool use. Builds on 15 of its 26 references that are in the map. First posted on arXiv 27 Feb 2026.
Builds on
15 of 26 references.
Closest ancestors · 4
Same direction, newest first
- 2025CudaForge: An Agent Framework with Hardware Feedback for CUDA Kernel Optimization23 Oct
- 2025STARK: Strategic Team of Agents for Refining Kernels19 Oct
- 2024OpenHands: An Open Platform for AI Software Developers as Generalist Agents23 Jul
- 2024SWE-agent: Agent-Computer Interfaces Enable Automated Software Engineering6 May
Other references · 11
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2022ReAct: Synergizing Reasoning and Acting in Language Modelscited by 8,355
- 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scalecited by 2,678
- 2015High-Dimensional Continuous Control Using Generalized Advantage Estimationcited by 1,916
- 2022The Stack: 3 TB of permissively licensed source codecited by 354
- 2025KernelBench: Can LLMs Write Efficient GPU Kernels?cited by 206
- 2025Kevin: Multi-Turn RL for Generating CUDA Kernelscited by 73
- 2025CUDA-L1: Improving CUDA Optimization via Contrastive Reinforcement Learningcited by 59
- 2025EvoEngineer: Mastering Automated CUDA Kernel Code Evolution with Large Language Modelscited by 13
- 2025ConCuR: Conciseness Makes State-of-the-Art Kernel Generationcited by 13
- 2025From Large to Small: Transferring CUDA Optimization Expertise via Reasoning Graphcited by 9
Built on by
Citing papers per year: 2026: 41.
41 papers.
Sort:
By area:
- 2026AutoKernel: Autonomous GPU Kernel Optimization via Iterative Agent-Driven Searchcited by 14
- 2026AI4AI-Bench: Benchmarking LLM Agents in Algorithmic Design for Recursive Self-Improvementcited by 10
- 2026Hardening Agent Benchmarks with Adversarial Hacker-Fixer Loopscited by 8
- 2026Heddle: A Distributed Orchestration System for Agentic RL Rolloutcited by 6
- 2026KernelBench-Verified: Do LLM-Generated Kernels Actually Beat PyTorch?cited by 6
- 2026Kernel-Smith: A Unified Recipe for Evolutionary Kernel Optimizationcited by 5
- 2026Improving Efficiency of GPU Kernel Optimization Agents using a Domain-Specific Language and Speed-of-Light Guidancecited by 4
- 2026InCoder-32B-Thinking: Industrial Code World Model for Thinkingcited by 4
- 2026Optimizing CUDA like a Human: Micro-Profiling Tools as Expert Surrogates for LLM-Based GPU Kernel Optimizationcited by 4
- 2026AdaExplore: Failure-Driven Adaptation and Diversity-Preserving Search for Efficient Kernel Generationcited by 3
- 2026daVinci-kernel: Co-Evolving Skill Selection, Summarization, and Utilization via RL for GPU Kernel Optimizationcited by 2
- 2026NKI-Agent: Domain-Specific Fine-Tuning and Agentic Tool Use for Neuron Kernel Generationcited by 2
29 more
- 2026CAKE: Compiler-Agent Co-Design for Frontier Kernel Evolutioncited by 2
- 2026CUDAHercules: Benchmarking Hardware-Aware Expert-level CUDA Optimization for LLMscited by 1
- 2026Prior Knowledge or Search? A Study of LLM Agents in Hardware-Aware Code Optimizationcited by 1
- 2026Xe-Forge: Multi-Stage LLM-Powered Kernel Optimization for Intel GPUcited by 1
- 2026PassNet: Scaling Large Language Models for Graph Compiler Pass Generationcited by 1
- 2026HTAM: Hierarchical Transition-Attended Memory for Operator Optimizationcited by 1
- 2026Hawk: Harnessing Hardware-Aware Knowledge for High-Performance NPU Kernel Generationcited by 1
- 2026MKEvolve: A Modular Multi-Agent Framework for Kernel Code Generationcited by 1
- 2026CANN Bench: Benchmarking Agent Generated Kernels against Real NPU and Algorithmic Limitscited by 1
- 2026Ave: Guiding Agentic GPU Optimization Using Data-Flow Invariantscited by 0
- 2026GPU Forecasters: Language Models as Selective Surrogates for Kernel Runtime Optimizationcited by 0
- 2026Multi-turn RL with Structural and Performance Aware Rewards for CUDA Kernel Generationcited by 0
- 2026KernelGenBench: Can LLMs and Agents Write Efficient Kernels Across Operator Sources and Hardware Platforms?cited by 0
- 2026WIDE: Boosting Adaptive LLM Inference via Token-level Dynamic Width Pruningcited by 0
- 2026LEAP: Lean Environment-Feedback via Adaptive Pruning for Code RL in GPU Kernel Generationcited by 0
- 2026KernelBrain: Coarse-to-Fine, Budget-Aware Search for Agentic GPU Kernel Optimizationcited by 0
- 2026Don't Regenerate, Debug: A Domain-Specific Agent for Repairing Near-Miss Hardware Operatorscited by 0
- 2026Anatomy of a Quantized Agent: VRAM Stability and Forecasting in Code-Synthesis Agentic Workloadscited by 0
- 2026PTXBench: Benchmarking and Adapting LLMs for GPU Kernel Optimization with Architecture-specific PTXcited by 0
- 2026AsmEvo: Agentic Assembly-Level Optimization of AMD GPU Kernels with Functional Equivalence Verificationcited by 0
- 2026DataKernelBench: Can LLMs Optimize Database Queries on GPUs?cited by 0
- 2026CUDA-Harness: Harnessing Agentic CUDA Kernel Generation and Optimization from Natural Languagecited by 0
- 2026AMDKernelVault: Large-Scale Datasets and Agentic Training for AMD GPU Kernel Optimizationcited by 0
- 2026FlashVector: Agent for Hierarchical Model Serving Stack Optimizationcited by 0
- 2026OSFoundry: Building and Evolving Operating Systems with Specification-Guided Agentscited by 0
- 2026CliffCompaction: Cost-Efficient Compaction for Long-Horizon Coding Agentscited by 0
- 2026BEHAVE: Functional Behavior Modeling Enables Self-Improving Agents for Hardware Design and Verificationcited by 0
- 2026Self-Evolving Coding Agents: From Digital Programs to Physical-World Intelligencecited by 0
- 2026From Experience to Expertise: Adoption-Aware Memory Learning for Data-Scarce NPU Kernel Synthesiscited by 0
- 2026From Experience to Expertise: Adoption-Aware Memory Learning for Data-Scarce NPU Kernel Synthesis28 Sep 2026
- 2026Self-Evolving Coding Agents: From Digital Programs to Physical-World Intelligence28 Sep 2026
- 2026BEHAVE: Functional Behavior Modeling Enables Self-Improving Agents for Hardware Design and Verification28 Sep 2026
- 2026CliffCompaction: Cost-Efficient Compaction for Long-Horizon Coding Agents22 Sep 2026
- 2026FlashVector: Agent for Hierarchical Model Serving Stack Optimization15 Sep 2026
- 2026AMDKernelVault: Large-Scale Datasets and Agentic Training for AMD GPU Kernel Optimization11 Sep 2026
- 2026CUDA-Harness: Harnessing Agentic CUDA Kernel Generation and Optimization from Natural Language30 Aug 2026
- 2026DataKernelBench: Can LLMs Optimize Database Queries on GPUs?25 Aug 2026
- 2026AsmEvo: Agentic Assembly-Level Optimization of AMD GPU Kernels with Functional Equivalence Verification21 Aug 2026
- 2026AI4AI-Bench: Benchmarking LLM Agents in Algorithmic Design for Recursive Self-Improvement20 Aug 2026
- 2026PTXBench: Benchmarking and Adapting LLMs for GPU Kernel Optimization with Architecture-specific PTX18 Aug 2026
- 2026Anatomy of a Quantized Agent: VRAM Stability and Forecasting in Code-Synthesis Agentic Workloads15 Aug 2026
29 more
- 2026CAKE: Compiler-Agent Co-Design for Frontier Kernel Evolution12 Aug 2026
- 2026OSFoundry: Building and Evolving Operating Systems with Specification-Guided Agents8 Aug 2026
- 2026Don't Regenerate, Debug: A Domain-Specific Agent for Repairing Near-Miss Hardware Operators3 Aug 2026
- 2026LEAP: Lean Environment-Feedback via Adaptive Pruning for Code RL in GPU Kernel Generation3 Aug 2026
- 2026WIDE: Boosting Adaptive LLM Inference via Token-level Dynamic Width Pruning30 Jul 2026
- 2026Multi-turn RL with Structural and Performance Aware Rewards for CUDA Kernel Generation23 Jul 2026
- 2026KernelGenBench: Can LLMs and Agents Write Efficient Kernels Across Operator Sources and Hardware Platforms?22 Jul 2026
- 2026CANN Bench: Benchmarking Agent Generated Kernels against Real NPU and Algorithmic Limits8 Jul 2026
- 2026NKI-Agent: Domain-Specific Fine-Tuning and Agentic Tool Use for Neuron Kernel Generation5 Jul 2026
- 2026Hawk: Harnessing Hardware-Aware Knowledge for High-Performance NPU Kernel Generation2 Jul 2026
- 2026KernelBench-Verified: Do LLM-Generated Kernels Actually Beat PyTorch?26 Jun 2026
- 2026Optimizing CUDA like a Human: Micro-Profiling Tools as Expert Surrogates for LLM-Based GPU Kernel Optimization24 Jun 2026
- 2026MKEvolve: A Modular Multi-Agent Framework for Kernel Code Generation20 Jun 2026
- 2026daVinci-kernel: Co-Evolving Skill Selection, Summarization, and Utilization via RL for GPU Kernel Optimization15 Jun 2026
- 2026Hardening Agent Benchmarks with Adversarial Hacker-Fixer Loops8 Jun 2026
- 2026GPU Forecasters: Language Models as Selective Surrogates for Kernel Runtime Optimization29 May 2026
- 2026HTAM: Hierarchical Transition-Attended Memory for Operator Optimization28 May 2026
- 2026PassNet: Scaling Large Language Models for Graph Compiler Pass Generation28 May 2026
- 2026KernelBrain: Coarse-to-Fine, Budget-Aware Search for Agentic GPU Kernel Optimization19 May 2026
- 2026Prior Knowledge or Search? A Study of LLM Agents in Hardware-Aware Code Optimization19 May 2026
- 2026CUDAHercules: Benchmarking Hardware-Aware Expert-level CUDA Optimization for LLMs8 May 2026
- 2026AdaExplore: Failure-Driven Adaptation and Diversity-Preserving Search for Efficient Kernel Generation17 Apr 2026
- 2026Xe-Forge: Multi-Stage LLM-Powered Kernel Optimization for Intel GPU16 Apr 2026
- 2026Ave: Guiding Agentic GPU Optimization Using Data-Flow Invariants16 Apr 2026
- 2026InCoder-32B-Thinking: Industrial Code World Model for Thinking3 Apr 2026
- 2026Improving Efficiency of GPU Kernel Optimization Agents using a Domain-Specific Language and Speed-of-Light Guidance30 Mar 2026
- 2026Kernel-Smith: A Unified Recipe for Evolutionary Kernel Optimization30 Mar 2026
- 2026Heddle: A Distributed Orchestration System for Agentic RL Rollout30 Mar 2026
- 2026AutoKernel: Autonomous GPU Kernel Optimization via Iterative Agent-Driven Search22 Mar 2026
- 2026AI4AI-Bench: Benchmarking LLM Agents in Algorithmic Design for Recursive Self-Improvementcited by 10
- 2026Hardening Agent Benchmarks with Adversarial Hacker-Fixer Loopscited by 8
- 2026Heddle: A Distributed Orchestration System for Agentic RL Rolloutcited by 6
- 2026Improving Efficiency of GPU Kernel Optimization Agents using a Domain-Specific Language and Speed-of-Light Guidancecited by 4
- 2026NKI-Agent: Domain-Specific Fine-Tuning and Agentic Tool Use for Neuron Kernel Generationcited by 2
- 2026Prior Knowledge or Search? A Study of LLM Agents in Hardware-Aware Code Optimizationcited by 1
- 2026MKEvolve: A Modular Multi-Agent Framework for Kernel Code Generationcited by 1
- 2026Ave: Guiding Agentic GPU Optimization Using Data-Flow Invariantscited by 0
- 2026GPU Forecasters: Language Models as Selective Surrogates for Kernel Runtime Optimizationcited by 0
- 2026KernelGenBench: Can LLMs and Agents Write Efficient Kernels Across Operator Sources and Hardware Platforms?cited by 0
- 2026Don't Regenerate, Debug: A Domain-Specific Agent for Repairing Near-Miss Hardware Operatorscited by 0
- 2026Anatomy of a Quantized Agent: VRAM Stability and Forecasting in Code-Synthesis Agentic Workloadscited by 0
12 most cited of 16 All 16 (JSON)
- 2026CliffCompaction: Cost-Efficient Compaction for Long-Horizon Coding Agents22 Sep 2026
- 2026AMDKernelVault: Large-Scale Datasets and Agentic Training for AMD GPU Kernel Optimization11 Sep 2026
- 2026CUDA-Harness: Harnessing Agentic CUDA Kernel Generation and Optimization from Natural Language30 Aug 2026
- 2026AI4AI-Bench: Benchmarking LLM Agents in Algorithmic Design for Recursive Self-Improvement20 Aug 2026
- 2026Anatomy of a Quantized Agent: VRAM Stability and Forecasting in Code-Synthesis Agentic Workloads15 Aug 2026
- 2026OSFoundry: Building and Evolving Operating Systems with Specification-Guided Agents8 Aug 2026
- 2026Don't Regenerate, Debug: A Domain-Specific Agent for Repairing Near-Miss Hardware Operators3 Aug 2026
- 2026KernelGenBench: Can LLMs and Agents Write Efficient Kernels Across Operator Sources and Hardware Platforms?22 Jul 2026
- 2026NKI-Agent: Domain-Specific Fine-Tuning and Agentic Tool Use for Neuron Kernel Generation5 Jul 2026
- 2026MKEvolve: A Modular Multi-Agent Framework for Kernel Code Generation20 Jun 2026
- 2026Hardening Agent Benchmarks with Adversarial Hacker-Fixer Loops8 Jun 2026
- 2026GPU Forecasters: Language Models as Selective Surrogates for Kernel Runtime Optimization29 May 2026
12 newest of 16 All 16 (JSON)
- 2026AutoKernel: Autonomous GPU Kernel Optimization via Iterative Agent-Driven Searchcited by 14
- 2026KernelBench-Verified: Do LLM-Generated Kernels Actually Beat PyTorch?cited by 6
- 2026Kernel-Smith: A Unified Recipe for Evolutionary Kernel Optimizationcited by 5
- 2026Optimizing CUDA like a Human: Micro-Profiling Tools as Expert Surrogates for LLM-Based GPU Kernel Optimizationcited by 4
- 2026AdaExplore: Failure-Driven Adaptation and Diversity-Preserving Search for Efficient Kernel Generationcited by 3
- 2026daVinci-kernel: Co-Evolving Skill Selection, Summarization, and Utilization via RL for GPU Kernel Optimizationcited by 2
- 2026CAKE: Compiler-Agent Co-Design for Frontier Kernel Evolutioncited by 2
- 2026CUDAHercules: Benchmarking Hardware-Aware Expert-level CUDA Optimization for LLMscited by 1
- 2026Xe-Forge: Multi-Stage LLM-Powered Kernel Optimization for Intel GPUcited by 1
- 2026PassNet: Scaling Large Language Models for Graph Compiler Pass Generationcited by 1
- 2026HTAM: Hierarchical Transition-Attended Memory for Operator Optimizationcited by 1
- 2026Hawk: Harnessing Hardware-Aware Knowledge for High-Performance NPU Kernel Generationcited by 1
12 most cited of 21 All 21 (JSON)
- 2026From Experience to Expertise: Adoption-Aware Memory Learning for Data-Scarce NPU Kernel Synthesis28 Sep 2026
- 2026FlashVector: Agent for Hierarchical Model Serving Stack Optimization15 Sep 2026
- 2026DataKernelBench: Can LLMs Optimize Database Queries on GPUs?25 Aug 2026
- 2026AsmEvo: Agentic Assembly-Level Optimization of AMD GPU Kernels with Functional Equivalence Verification21 Aug 2026
- 2026PTXBench: Benchmarking and Adapting LLMs for GPU Kernel Optimization with Architecture-specific PTX18 Aug 2026
- 2026CAKE: Compiler-Agent Co-Design for Frontier Kernel Evolution12 Aug 2026
- 2026LEAP: Lean Environment-Feedback via Adaptive Pruning for Code RL in GPU Kernel Generation3 Aug 2026
- 2026WIDE: Boosting Adaptive LLM Inference via Token-level Dynamic Width Pruning30 Jul 2026
- 2026CANN Bench: Benchmarking Agent Generated Kernels against Real NPU and Algorithmic Limits8 Jul 2026
- 2026Hawk: Harnessing Hardware-Aware Knowledge for High-Performance NPU Kernel Generation2 Jul 2026
- 2026KernelBench-Verified: Do LLM-Generated Kernels Actually Beat PyTorch?26 Jun 2026
- 2026Optimizing CUDA like a Human: Micro-Profiling Tools as Expert Surrogates for LLM-Based GPU Kernel Optimization24 Jun 2026
12 newest of 21 All 21 (JSON)
Similar, not cited
Close in wording, no citation either way
- 2026KernelBlaster: Continual Cross-Task CUDA Optimization via Memory-Augmented In-Context Reinforcement Learning
- 2026Kernel Forge: An Agent Harness for LLM-based Generation and Optimization of CUDA Kernels
- 2025cuPilot: A Strategy-Coordinated Multi-agent Framework for CUDA Kernel Evolution
- 2026Agentic Kernel Optimization: Generating State-of-the-Art GPU Kernels Without Hand-Written CUDA
- 2025Towards Robust Agentic CUDA Kernel Benchmarking, Verification, and Optimization
- 2025Astra: A Multi-Agent System for GPU Kernel Performance Optimization
- 2026From Tokens to Regions: CUDA-Sensitive Instruction Tuning for GPU Kernel Generation
- 2026AgentKernelArena: Generalization-Aware Benchmarking of GPU Kernel Optimization Agents