Open a request
Share a reproducible FlashInfer Trace definition and explain why the kernel matters.
Need a CUDA kernel for your training/inference but cannot find the experts to implement?
Submit your definitions and workloads to KDA-wishlist, and KDA will automatically optimize it w/ agentic workflows!
Open process. Public results. Built for real workloads.
Definition · Workloads · Target
Explore and benchmark candidates
Code · Benchmarks · Reproduction
CUDA kernels are foundational to the modern LLM ecosystem, but building high-performance implementations requires deep expertise.
KDA Wishlist is an experimental program where the KDA team uses agentic workflows to build the kernels the community needs.
HOW IT WORKS
Kernel Design Agents (KDA) are agentic CUDA systems that research, implement, verify, and iterate on performance-sensitive kernel tasks. Each request gives them a measurable contract—not just an idea.
Explore the KDA tech reportShare a reproducible FlashInfer Trace definition and explain why the kernel matters.
Others add a thumbs-up reaction to the GitHub issue and contribute useful workload context.
Accepted tasks enter a measured loop of analysis, generation, correctness checks, and profiling.
We publish the strongest validated kernel with benchmarks, reproduction steps, and known limits.
SELECTED ACHIEVEMENTS
KDA has already turned optimization targets into production contributions across LLM serving, image generation, and video workloads.
MLSys’26 FlashInfer Kernel Contest · B300 · MoE 2.25× · DSA 29.95× · GDN 6.10×
L1 Single Operation track · score 0.7608 · previous best 0.7584
MLSys’26 FlashInfer Contest · B300 · MoE 0.67× · DSA 11.91× · GDN 1.16×
KDA speedup for the Wan 2.2 workload on B200
10.621 → 5.240 ms weighted; model E2E +2.11%
B200 · 4.22–7.34× across 14 production shapes; E2E −9.16%
Results vary by hardware, model, and workload. See each linked pull request or repository for validation and reproduction details.
The workflow behind KDA earned 1st, 2nd, and 3rd place across the three fully agent-driven tracks in the MLSys 2026 FlashInfer contest.Review the released workflow and results
GOOD TO KNOW
Open the public wishlist, find a request you care about, and add a thumbs-up reaction to the top-level GitHub issue. Comments are best used for new workload evidence or implementation context.
Yes—unless we agree otherwise in advance. For accepted requests, KDA may publish the submitted definitions, generated implementations, benchmarks, profiling data, and design notes. We credit submitters for their definitions and workloads. If your business requires private kernels, email Ligeng Zhu at ligengz@nvidia.com to discuss a collaboration.
KDA typically delivers one to three of the most-requested kernels each week, depending on the team’s development workload. Sihao Liu and Ligeng Zhu currently cover the GPU and model-token costs personally, so capacity is limited. If you have a larger operator roadmap, email Ligeng Zhu at ligengz@nvidia.com to discuss establishing an official collaboration with NVIDIA.
Requests may need revision when they cannot be reproduced, lack a reliable correctness reference, target unsupported hardware, are too vague for automated evaluation, or have unclear licensing.
YOUR BOTTLENECK, NEXT
Bring the definition and real workloads. We’ll bring the research loop.
Submit a kernel request Or browse the wishlist and upvote