FlashInfer-Bench: Building the Virtuous Cycle for AI-driven LLM Systems
Not on the map. Why?
Abstract (arXiv)
Recent advances show that large language models (LLMs) can act as autonomous agents capable of generating GPU kernels, but integrating these AI-generated kernels into real-world inference systems remains challenging. FlashInfer-Bench addresses this gap by establishing a standardized, closed-loop framework that connects kernel generation, benchmarking, and deployment. At its core, FlashInfer Trace provides a unified schema describing kernel definitions, workloads, implementations, and evaluations, enabling consistent communication between agents and systems. Built on real serving traces, FlashInfer-Bench includes a curated dataset, a robust correctness- and performance-aware benchmarking framework, a public leaderboard to track LLM agents' GPU programming capabilities, and a dynamic substitution mechanism (apply()) that seamlessly injects the best-performing kernels into production LLM engines such as SGLang and vLLM. Using FlashInfer-Bench, we further evaluate the performance and limitations of LLM agents, compare the trade-offs among different GPU programming languages, and provide insights for future agent design. FlashInfer-Bench thus establishes a practical, reproducible pathway for continuously improving AI-generated kernels and deploying them into large-scale LLM inference.
Key facts. Cited by 20 papers in the map (data through 7 Oct 2026). Builds on 10 of its 23 references that are in the map. First posted on arXiv 1 Jan 2026.
Builds on
10 of 23 references.
References · 10
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,767
- 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
- 2023SGLang: Efficient Execution of Structured Language Model Programscited by 1,107
- 2019Fast Transformer Decoding: One Write-Head is All You Needcited by 700
- 2024FlashAttention-3: Fast and Accurate Attention with Asynchrony and Low-precisioncited by 507
- 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
- 2025Kevin: Multi-Turn RL for Generating CUDA Kernelscited by 73
- 2025TritonBench: Benchmarking Large Language Model Capabilities for Generating Triton Operatorscited by 68
Built on by
Citing papers per year: 2026: 20.
20 papers.
Sort:
By area:
- 2026K-Search: LLM Kernel Generation via Co-Evolving Intrinsic World Modelcited by 17
- 2026SOL-ExecBench: Speed-of-Light Benchmarking for Real-World GPU Kernels Against Hardware Limitscited by 13
- 2026Towards Automated Kernel Generation in the Era of LLMscited by 9
- 2026InferenceBench: A Benchmark for Open-Ended LLM Inference Optimization by AI Agentscited by 4
- 2026AdaExplore: Failure-Driven Adaptation and Diversity-Preserving Search for Efficient Kernel Generationcited by 3
- 2026FastKernels: Benchmarking GPU Kernel Generation in Productioncited by 2
- 2026Are LLM-Generated GPU Kernels Production-Ready? A Trace-Driven Benchmark and Optimization Agentcited by 2
- 2026DICE: Diffusion Large Language Models Excel at Generating CUDA Kernelscited by 1
- 2026Harness Engineering for LLM-Driven GPU Kernel Generationcited by 1
- 2026JAXBench: Benchmarking Autonomous TPU Kernel Optimizationcited by 1
- 2026MKEvolve: A Modular Multi-Agent Framework for Kernel Code Generationcited by 1
- 2026Kernel Forge: An Agent Harness for LLM-based Generation and Optimization of CUDA Kernelscited by 1
8 more
- 2026RealisticTritonBench: A Benchmark for Triton-Kernel Generation in Real-World AI Frameworkscited by 1
- 2026Towards Feedback-to-Plan Decisions for Self-Evolving LLM Agents in CUDA Kernel Generationcited by 0
- 2026PithTrain: A Compact and Agent-Native MoE Training Systemcited by 0
- 2026KernelGenBench: Can LLMs and Agents Write Efficient Kernels Across Operator Sources and Hardware Platforms?cited by 0
- 2026PTXBench: Benchmarking and Adapting LLMs for GPU Kernel Optimization with Architecture-specific PTXcited by 0
- 2026RoofLang: Enabling AI-Driven Architecting of LLM Inference Systemscited by 0
- 2026When Agents Slow Down: Understanding LLM Agents' Test-Time Strategies via Elo-per-token Analysiscited by 0
- 2026EasyPPO: Stabilizing the Critic Is Keycited by 0
- 2026EasyPPO: Stabilizing the Critic Is Key29 Sep 2026
- 2026When Agents Slow Down: Understanding LLM Agents' Test-Time Strategies via Elo-per-token Analysis14 Sep 2026
- 2026RoofLang: Enabling AI-Driven Architecting of LLM Inference Systems11 Sep 2026
- 2026PTXBench: Benchmarking and Adapting LLMs for GPU Kernel Optimization with Architecture-specific PTX18 Aug 2026
- 2026RealisticTritonBench: A Benchmark for Triton-Kernel Generation in Real-World AI Frameworks12 Aug 2026
- 2026KernelGenBench: Can LLMs and Agents Write Efficient Kernels Across Operator Sources and Hardware Platforms?22 Jul 2026
- 2026Harness Engineering for LLM-Driven GPU Kernel Generation20 Jul 2026
- 2026Are LLM-Generated GPU Kernels Production-Ready? A Trace-Driven Benchmark and Optimization Agent16 Jul 2026
- 2026MKEvolve: A Modular Multi-Agent Framework for Kernel Code Generation20 Jun 2026
- 2026Kernel Forge: An Agent Harness for LLM-based Generation and Optimization of CUDA Kernels2 Jun 2026
- 2026PithTrain: A Compact and Agent-Native MoE Training System29 May 2026
- 2026Towards Feedback-to-Plan Decisions for Self-Evolving LLM Agents in CUDA Kernel Generation26 May 2026
8 more
- 2026FastKernels: Benchmarking GPU Kernel Generation in Production22 May 2026
- 2026InferenceBench: A Benchmark for Open-Ended LLM Inference Optimization by AI Agents20 May 2026
- 2026JAXBench: Benchmarking Autonomous TPU Kernel Optimization19 May 2026
- 2026AdaExplore: Failure-Driven Adaptation and Diversity-Preserving Search for Efficient Kernel Generation17 Apr 2026
- 2026SOL-ExecBench: Speed-of-Light Benchmarking for Real-World GPU Kernels Against Hardware Limits19 Mar 2026
- 2026K-Search: LLM Kernel Generation via Co-Evolving Intrinsic World Model22 Feb 2026
- 2026DICE: Diffusion Large Language Models Excel at Generating CUDA Kernels12 Feb 2026
- 2026Towards Automated Kernel Generation in the Era of LLMs22 Jan 2026
- 2026SOL-ExecBench: Speed-of-Light Benchmarking for Real-World GPU Kernels Against Hardware Limitscited by 13
- 2026Towards Automated Kernel Generation in the Era of LLMscited by 9
- 2026InferenceBench: A Benchmark for Open-Ended LLM Inference Optimization by AI Agentscited by 4
- 2026FastKernels: Benchmarking GPU Kernel Generation in Productioncited by 2
- 2026MKEvolve: A Modular Multi-Agent Framework for Kernel Code Generationcited by 1
- 2026Kernel Forge: An Agent Harness for LLM-based Generation and Optimization of CUDA Kernelscited by 1
- 2026Towards Feedback-to-Plan Decisions for Self-Evolving LLM Agents in CUDA Kernel Generationcited by 0
- 2026PithTrain: A Compact and Agent-Native MoE Training Systemcited by 0
- 2026KernelGenBench: Can LLMs and Agents Write Efficient Kernels Across Operator Sources and Hardware Platforms?cited by 0
- 2026KernelGenBench: Can LLMs and Agents Write Efficient Kernels Across Operator Sources and Hardware Platforms?22 Jul 2026
- 2026MKEvolve: A Modular Multi-Agent Framework for Kernel Code Generation20 Jun 2026
- 2026Kernel Forge: An Agent Harness for LLM-based Generation and Optimization of CUDA Kernels2 Jun 2026
- 2026PithTrain: A Compact and Agent-Native MoE Training System29 May 2026
- 2026Towards Feedback-to-Plan Decisions for Self-Evolving LLM Agents in CUDA Kernel Generation26 May 2026
- 2026FastKernels: Benchmarking GPU Kernel Generation in Production22 May 2026
- 2026InferenceBench: A Benchmark for Open-Ended LLM Inference Optimization by AI Agents20 May 2026
- 2026SOL-ExecBench: Speed-of-Light Benchmarking for Real-World GPU Kernels Against Hardware Limits19 Mar 2026
- 2026Towards Automated Kernel Generation in the Era of LLMs22 Jan 2026
- 2026AdaExplore: Failure-Driven Adaptation and Diversity-Preserving Search for Efficient Kernel Generationcited by 3
- 2026Are LLM-Generated GPU Kernels Production-Ready? A Trace-Driven Benchmark and Optimization Agentcited by 2
- 2026DICE: Diffusion Large Language Models Excel at Generating CUDA Kernelscited by 1
- 2026Harness Engineering for LLM-Driven GPU Kernel Generationcited by 1
- 2026JAXBench: Benchmarking Autonomous TPU Kernel Optimizationcited by 1
- 2026RealisticTritonBench: A Benchmark for Triton-Kernel Generation in Real-World AI Frameworkscited by 1
- 2026PTXBench: Benchmarking and Adapting LLMs for GPU Kernel Optimization with Architecture-specific PTXcited by 0
- 2026EasyPPO: Stabilizing the Critic Is Keycited by 0
- 2026EasyPPO: Stabilizing the Critic Is Key29 Sep 2026
- 2026PTXBench: Benchmarking and Adapting LLMs for GPU Kernel Optimization with Architecture-specific PTX18 Aug 2026
- 2026RealisticTritonBench: A Benchmark for Triton-Kernel Generation in Real-World AI Frameworks12 Aug 2026
- 2026Harness Engineering for LLM-Driven GPU Kernel Generation20 Jul 2026
- 2026Are LLM-Generated GPU Kernels Production-Ready? A Trace-Driven Benchmark and Optimization Agent16 Jul 2026
- 2026JAXBench: Benchmarking Autonomous TPU Kernel Optimization19 May 2026
- 2026AdaExplore: Failure-Driven Adaptation and Diversity-Preserving Search for Efficient Kernel Generation17 Apr 2026
- 2026DICE: Diffusion Large Language Models Excel at Generating CUDA Kernels12 Feb 2026
Similar, not cited
Close in wording, no citation either way
- 2025STARK: Strategic Team of Agents for Refining Kernels
- 2026KernelSight-LM: A Kernel-Level LLM Inference Simulator
- 2026KernelCraft: Benchmarking for Agentic Close-to-Metal Kernel Generation on Emerging Hardware
- 2026LLM4LLM: Bridging Kernel Benchmarks and Real Deployment via Closed-Loop Agentic Optimization
- 2026Characterizing Parallelism Strategies in LLM Inference: Fundamental Compute-Communication Trade-offs
- 2026KForge: LLM-Driven Cross-Platform Kernel Generation for AI Accelerators
- 2025Astra: A Multi-Agent System for GPU Kernel Performance Optimization
- 2024Demystifying AI Platform Design for Distributed Inference of Next-Generation LLM models