AccelOpt: A Self-Improving LLM Agentic System for AI Accelerator Kernel Optimization
On the map: LLM agents and tool use → Software engineering coding agents
also in 1 more
Abstract (arXiv)
We present AccelOpt, a self-improving large language model (LLM) agentic system that autonomously optimizes kernels for emerging AI acclerators, eliminating the need for expert-provided hardware-specific optimization knowledge. AccelOpt explores the kernel optimization space through iterative generation, informed by an optimization memory that curates experiences and insights from previously encountered slow-fast kernel pairs. We build NKIBench, a new benchmark suite of AWS Trainium accelerator kernels with varying complexity extracted from real-world LLM workloads to evaluate the effectiveness of AccelOpt. Our evaluation confirms that AccelOpt's capability improves over time, boosting the average percentage of peak throughput from 49% to 61% on Trainium 1 and from 45% to 59% on Trainium 2 for NKIBench kernels. Moreover, AccelOpt is highly cost-effective: using open-source models, it matches the kernel improvements of Claude Sonnet 4 while being 26× cheaper. The code is open-sourced at https://github.com/zhang677/AccelOpt.
Key facts. Cited by 21 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 35 of its 63 references that are in the map. First posted on arXiv 19 Nov 2025.
Builds on
35 of 63 references.
Closest ancestors · 3
Same direction, newest first
- 2025Towards Robust Agentic CUDA Kernel Benchmarking, Verification, and Optimization16 Sep
- 2025AlphaEvolve: A coding agent for scientific and algorithmic discovery16 Jun
- 2025Adaptive Self-improvement LLM Agentic System for ML Library Development4 Feb
Other references · 32
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,767
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,616
- 2023Reflexion: Language Agents with Verbal Reinforcement Learningcited by 4,064
- 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,551
- 2024Large Language Monkeys: Scaling Inference Compute with Repeated Samplingcited by 946
- 2024DeepSeekMoE: Towards Ultimate Expert Specialization in Mixture-of-Experts Language Modelscited by 834
- 2024A Survey on the Memory Mechanism of Large Language Model based Agentscited by 622
- 2023The Falcon Series of Open Language Modelscited by 526
- 2024FlashAttention-3: Fast and Accurate Attention with Asynchrony and Low-precisioncited by 507
- 2025GEPA: Reflective Prompt Evolution Can Outperform Reinforcement Learningcited by 449
- 2025Agentic Context Engineering: Evolving Contexts for Self-Improving Language Modelscited by 349
- 2024Learning to (Learn at Test Time): RNNs with Expressive Hidden Statescited by 329
20 more
- 2025ReasoningBank: Scaling Agent Self-Evolving with Reasoning Memorycited by 253
- 2025FlashInfer: Efficient and Customizable Attention Engine for LLM Inference Servingcited by 248
- 2025KernelBench: Can LLMs Write Efficient GPU Kernels?cited by 206
- 2020Ansor: Generating High-Performance Tensor Programs for Deep Learningcited by 199
- 2023TPU v4: An Optically Reconfigurable Supercomputer for Machine Learning with Hardware Support for Embeddingscited by 198
- 2023Full Stack Optimization of Transformer Inference: a Surveycited by 79
- 2025Kevin: Multi-Turn RL for Generating CUDA Kernelscited by 73
- 2025SEAgent: Self-Evolving Computer Use Agent with Autonomous Learning from Experiencecited by 68
- 2025CUDA-L1: Improving CUDA Optimization via Contrastive Reinforcement Learningcited by 59
- 2024Mirage: A Multi-Level Superoptimizer for Tensor Programscited by 39
- 2025Insights into DeepSeek-V3: Scaling Challenges and Reflections on Hardware for AI Architecturescited by 38
- 2025MultiKernelBench: A Multi-Platform Benchmark for Kernel Generationcited by 35
- 2024ThunderKittens: Simple, Fast, and Adorable AI Kernelscited by 34
- 2025TritonRL: Training LLMs to Think and Code Triton Without Cheatingcited by 24
- 2025From Few to Many: Self-Improving Many-Shot Reasoners Through Iterative Optimization and Generationcited by 20
- 2024SambaNova SN40L: Scaling the AI Memory Wall with Dataflow and Composition of Expertscited by 18
- 2024Improving Parallel Program Performance with LLM Optimizers via Agent-System Interfacescited by 11
- 2025Dato: A Task-Based Programming Model for Dataflow Acceleratorscited by 5
- 2022Stardust: Compiling Sparse Tensor Algebra to a Reconfigurable Dataflow Architecturecited by 3
- 2025TTrace: Lightweight Error Checking and Diagnosis for Distributed Trainingcited by 1
Built on by
Citing papers per year: 2026: 21.
21 papers.
Sort:
By area:
- 2026From Storage to Experience: A Survey on the Evolution of LLM Agent Memory Mechanismscited by 19
- 2026Understanding the Challenges in Iterative Generative Optimization with LLMscited by 8
- 2026AdaExplore: Failure-Driven Adaptation and Diversity-Preserving Search for Efficient Kernel Generationcited by 3
- 2026From Human Guidance to Autonomy: Agent Skill System for End-to-End LLM Deployment on Spatial NPUscited by 2
- 2026NKI-Agent: Domain-Specific Fine-Tuning and Agentic Tool Use for Neuron Kernel Generationcited by 2
- 2026PerfAgent: Profiler-Guided Iterative Refinement for Repository-Level Code Optimizationcited by 2
- 2026Architectural Implications of Agentic AI Workflowscited by 2
- 2026CAKE: Compiler-Agent Co-Design for Frontier Kernel Evolutioncited by 2
- 2026Surprisal-Guided Selection: Compute-Optimal Test-Time Strategies for Execution-Grounded Code Generationcited by 1
- 2026RuleFlow : Generating Reusable Program Optimizations with LLMscited by 1
- 2026MKEvolve: A Modular Multi-Agent Framework for Kernel Code Generationcited by 1
- 2026Learning When to Optimize: Verified Optimization Skills from Expert GPU-Kernel Lineagescited by 0
9 more
- 2026Arbor: Tree Search as a Cognition Layer for Autonomous Agentscited by 0
- 2026IDP AutoOpt: Agent-Driven Optimization of Document Processing Pipeline Configurationscited by 0
- 2026KernelArc: A Multi-Agent Framework for GPU Kernel Optimizationcited by 0
- 2026PTXBench: Benchmarking and Adapting LLMs for GPU Kernel Optimization with Architecture-specific PTXcited by 0
- 2026From PyTorch to the NPU: LLM-Agent-Driven Model Conversion Across Heterogeneous Inference Runtimescited by 0
- 2026KernelOPT: Dispatch-Aware Agentic Search for GPU Kernel Optimizationcited by 0
- 2026Sol-H3: Recursive Self-Improvement for MiniMax-H3 Inference Acceleration on Sol-Engine across Cloud and Edgecited by 0
- 2026From Experience to Expertise: Adoption-Aware Memory Learning for Data-Scarce NPU Kernel Synthesiscited by 0
- 2026GPUPhysBench: Benchmarking Coding Agents for Correct and Efficient GPU Physics Simulationcited by 0
- 2026GPUPhysBench: Benchmarking Coding Agents for Correct and Efficient GPU Physics Simulation28 Sep 2026
- 2026From Experience to Expertise: Adoption-Aware Memory Learning for Data-Scarce NPU Kernel Synthesis28 Sep 2026
- 2026Sol-H3: Recursive Self-Improvement for MiniMax-H3 Inference Acceleration on Sol-Engine across Cloud and Edge28 Sep 2026
- 2026KernelOPT: Dispatch-Aware Agentic Search for GPU Kernel Optimization24 Sep 2026
- 2026From PyTorch to the NPU: LLM-Agent-Driven Model Conversion Across Heterogeneous Inference Runtimes23 Sep 2026
- 2026PTXBench: Benchmarking and Adapting LLMs for GPU Kernel Optimization with Architecture-specific PTX18 Aug 2026
- 2026KernelArc: A Multi-Agent Framework for GPU Kernel Optimization17 Aug 2026
- 2026CAKE: Compiler-Agent Co-Design for Frontier Kernel Evolution12 Aug 2026
- 2026Architectural Implications of Agentic AI Workflows5 Aug 2026
- 2026PerfAgent: Profiler-Guided Iterative Refinement for Repository-Level Code Optimization22 Jul 2026
- 2026IDP AutoOpt: Agent-Driven Optimization of Document Processing Pipeline Configurations13 Jul 2026
- 2026NKI-Agent: Domain-Specific Fine-Tuning and Agentic Tool Use for Neuron Kernel Generation5 Jul 2026
9 more
- 2026MKEvolve: A Modular Multi-Agent Framework for Kernel Code Generation20 Jun 2026
- 2026Arbor: Tree Search as a Cognition Layer for Autonomous Agents10 Jun 2026
- 2026From Human Guidance to Autonomy: Agent Skill System for End-to-End LLM Deployment on Spatial NPUs27 May 2026
- 2026Learning When to Optimize: Verified Optimization Skills from Expert GPU-Kernel Lineages27 May 2026
- 2026From Storage to Experience: A Survey on the Evolution of LLM Agent Memory Mechanisms7 May 2026
- 2026AdaExplore: Failure-Driven Adaptation and Diversity-Preserving Search for Efficient Kernel Generation17 Apr 2026
- 2026Understanding the Challenges in Iterative Generative Optimization with LLMs25 Mar 2026
- 2026Surprisal-Guided Selection: Compute-Optimal Test-Time Strategies for Execution-Grounded Code Generation7 Feb 2026
- 2026RuleFlow : Generating Reusable Program Optimizations with LLMs6 Feb 2026
- 2026From Storage to Experience: A Survey on the Evolution of LLM Agent Memory Mechanismscited by 19
- 2026From Human Guidance to Autonomy: Agent Skill System for End-to-End LLM Deployment on Spatial NPUscited by 2
- 2026NKI-Agent: Domain-Specific Fine-Tuning and Agentic Tool Use for Neuron Kernel Generationcited by 2
- 2026Architectural Implications of Agentic AI Workflowscited by 2
- 2026MKEvolve: A Modular Multi-Agent Framework for Kernel Code Generationcited by 1
- 2026Learning When to Optimize: Verified Optimization Skills from Expert GPU-Kernel Lineagescited by 0
- 2026Arbor: Tree Search as a Cognition Layer for Autonomous Agentscited by 0
- 2026IDP AutoOpt: Agent-Driven Optimization of Document Processing Pipeline Configurationscited by 0
- 2026From PyTorch to the NPU: LLM-Agent-Driven Model Conversion Across Heterogeneous Inference Runtimescited by 0
- 2026KernelOPT: Dispatch-Aware Agentic Search for GPU Kernel Optimizationcited by 0
- 2026KernelOPT: Dispatch-Aware Agentic Search for GPU Kernel Optimization24 Sep 2026
- 2026From PyTorch to the NPU: LLM-Agent-Driven Model Conversion Across Heterogeneous Inference Runtimes23 Sep 2026
- 2026Architectural Implications of Agentic AI Workflows5 Aug 2026
- 2026IDP AutoOpt: Agent-Driven Optimization of Document Processing Pipeline Configurations13 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
- 2026Arbor: Tree Search as a Cognition Layer for Autonomous Agents10 Jun 2026
- 2026From Human Guidance to Autonomy: Agent Skill System for End-to-End LLM Deployment on Spatial NPUs27 May 2026
- 2026Learning When to Optimize: Verified Optimization Skills from Expert GPU-Kernel Lineages27 May 2026
- 2026From Storage to Experience: A Survey on the Evolution of LLM Agent Memory Mechanisms7 May 2026
- 2026Understanding the Challenges in Iterative Generative Optimization with LLMscited by 8
- 2026AdaExplore: Failure-Driven Adaptation and Diversity-Preserving Search for Efficient Kernel Generationcited by 3
- 2026PerfAgent: Profiler-Guided Iterative Refinement for Repository-Level Code Optimizationcited by 2
- 2026CAKE: Compiler-Agent Co-Design for Frontier Kernel Evolutioncited by 2
- 2026Surprisal-Guided Selection: Compute-Optimal Test-Time Strategies for Execution-Grounded Code Generationcited by 1
- 2026RuleFlow : Generating Reusable Program Optimizations with LLMscited by 1
- 2026KernelArc: A Multi-Agent Framework for GPU Kernel Optimizationcited by 0
- 2026PTXBench: Benchmarking and Adapting LLMs for GPU Kernel Optimization with Architecture-specific PTXcited by 0
- 2026From Experience to Expertise: Adoption-Aware Memory Learning for Data-Scarce NPU Kernel Synthesiscited by 0
- 2026GPUPhysBench: Benchmarking Coding Agents for Correct and Efficient GPU Physics Simulationcited by 0
- 2026GPUPhysBench: Benchmarking Coding Agents for Correct and Efficient GPU Physics Simulation28 Sep 2026
- 2026From Experience to Expertise: Adoption-Aware Memory Learning for Data-Scarce NPU Kernel Synthesis28 Sep 2026
- 2026PTXBench: Benchmarking and Adapting LLMs for GPU Kernel Optimization with Architecture-specific PTX18 Aug 2026
- 2026KernelArc: A Multi-Agent Framework for GPU Kernel Optimization17 Aug 2026
- 2026CAKE: Compiler-Agent Co-Design for Frontier Kernel Evolution12 Aug 2026
- 2026PerfAgent: Profiler-Guided Iterative Refinement for Repository-Level Code Optimization22 Jul 2026
- 2026AdaExplore: Failure-Driven Adaptation and Diversity-Preserving Search for Efficient Kernel Generation17 Apr 2026
- 2026Understanding the Challenges in Iterative Generative Optimization with LLMs25 Mar 2026
- 2026Surprisal-Guided Selection: Compute-Optimal Test-Time Strategies for Execution-Grounded Code Generation7 Feb 2026
- 2026RuleFlow : Generating Reusable Program Optimizations with LLMs6 Feb 2026
Similar, not cited
Close in wording, no citation either way
- 2026KernelGenBench: Can LLMs and Agents Write Efficient Kernels Across Operator Sources and Hardware Platforms?
- 2026KernelCraft: Benchmarking for Agentic Close-to-Metal Kernel Generation on Emerging Hardware
- 2025STARK: Strategic Team of Agents for Refining Kernels
- 2026Towards Automated Kernel Generation in the Era of LLMs
- 2026LLM4LLM: Bridging Kernel Benchmarks and Real Deployment via Closed-Loop Agentic Optimization
- 2026KForge: LLM-Driven Cross-Platform Kernel Generation for AI Accelerators
- 2025ASAP: an Agentic Solution to Auto-optimize Performance of Large-Scale LLM Training
- 2026AI Hardware Accelerators for Large Language Models: Architectures and the Memory Wall ↗