WorksComputer Science
JustGoal: With Only a Goal, the AI Baseline Is Above the Expert Kernels
Abstract
A small number of GPU kernels often determines the efficiency of large language model training and inference. High-performance kernel development typically relies on expert knowledge through manual implementation, collaboration with general-purpose models, or specialized agents equipped with domain knowledge, evaluation methods, and search workflows. We investigate how much value this knowledge provides as foundation models become stronger. We introduce JustGoal, a goal-only baseline in which a general-purpose agent receives only the optimization objective and constraints, without kernel-specific infrastructure, knowledge bases, or predefined workflows, and builds its own evaluation loop. On NVIDIA B200 GPUs, JustGoal outperforms open-source libraries on 8 of 10 operators, achieving per-operator geometric mean speedups of 1.14x to 2.34x. It also surpasses the fastest publicly released specialized-agent kernel on 5 of 9 operators with available comparisons. In a controlled same-model comparison using Fable 5.1 on single shapes, the KDA-0.5 harness reaches 0.95x of JustGoal's performance, while single-mechanism ablations show no consistent benefit from domain knowledge or predefined workflows. These results indicate that specialized machinery should be justified by clear gains over a same-model, same-scope goal-only baseline and reevaluated whenever the foundation model changes.
Introduction
The computational efficiency of large language model training and inference depends critically on a small set of GPU kernels. Developing high-performance implementations of these kernels requires jointly optimizing tiling strategies, memory hierarchy utilization, asynchronous execution pipelines, and hardware-specific instructions. These design choices are tightly coupled, and their optimal configuration changes across hardware generations. Consequently, the development of high-performance kernels has long relied on leading experts devoting weeks or even months to manual optimization. The FlashAttention series exemplifies this recurring investment: achieving high performance on successive GPU architectures has required substantial redesigns of its algorithms and execution pipelines [6, 20, 7]. Large language models have recently entered kernel development. Experts may collaborate directly with general-purpose models. They may also encode evaluation methods, domain knowledge, and search workflows into specialized kernel agents such as KDA, AVO, AKO, and CAKE. These agents have matched or surpassed expert-written implementations on selected operators [15, 16, 1, 22]. What these approaches share is their reliance on expert knowledge. As foundation models improve, capabilities that previously had to be supplied explicitly may become native to the model. This raises two related questions: how much additional value does expert knowledge continue to provide for kernel optimization as model capabilities increase, and how far can a model optimize kernels when given only the task objective and constraints, with as little predefined expert knowledge as possible?
To examine these questions, we construct a goal-only baseline and call its concrete experimental configuration JustGoal. It provides no kernel-specific evaluation infrastructure, domain knowledge, or search workflow. A general-purpose agent receives only the task objective and constraints. It builds its own evaluation loop and optimization process. We observe the role of expert knowledge from two measurable perspectives. First, we compare kernels generated by JustGoal with human-maintained open-source libraries and publicly released specialized-agent kernels, asking what performance level a goal-only configuration already reaches. Second, under the same model and task scope, we compare JustGoal with a specialized harness that adds evaluation infrastructure, domain knowledge, and search workflows, directly measuring the marginal value of these explicit expert mechanisms. Our experiments cover 11 operators spanning attention, linear attention, sparse attention, and mixture-of-experts (MoE). We independently verify every kernel in the comparison.
From the first perspective, JustGoal is faster than human-maintained open-source libraries on 8 of the 10 operators entering the comparison, and faster than the best publicly released specialized-agent kernel on 5 of the 9 operators that have one. This marks a capability threshold: without task-time injection of kernel-specific knowledge or optimization workflows, current frontier models can construct evaluation feedback, identify performance bottlenecks, and surpass human-maintained implementations on a range of operators. From the second perspective, in the scope-matched Fable 5.1 comparison, the full KDA-0.5 harness produces no overall gain on single-shape tasks, and single-mechanism ablations show no consistent gain from domain knowledge or predefined workflows. When the goal expands from optimizing one shape to covering every workload in an operator definition, the harness still provides a gain. To measure the net gain that the full specialized harness adds beyond the model’s own capabilities, it must be compared with a goal-only baseline under the same model and task scope. This net gain should be reevaluated whenever the foundation model changes.
Related Work
High-Performance Kernel Design
High-performance kernels must be co-designed around a model architecture and target hardware. On the one hand, even when the operator remains unchanged, changes in hardware resources and execution mechanisms can make an existing implementation inefficient. On the other hand, evolving model architectures introduce new operators and computational patterns. These two lines of evolution jointly shape kernel design and continually create new development and optimization demands.
Redesign driven by hardware evolution. High-performance kernel design requires aligning an algorithm’s computational and data movement demands with the resources of the target hardware. The FlashAttention series exemplifies this hardware-algorithm co-evolution: FlashAttention uses tiling to reduce data movement between high-bandwidth memory (HBM) and on-chip SRAM [6]; FlashAttention-3 reorganizes computation and data transfers around Hopper’s asynchronous execution capabilities [20]; and FlashAttention-4 further adapts the algorithm and execution pipeline to changes in the balance of hardware resources, addressing bottlenecks such as exponential evaluation and shared-memory access [7]. This progression illustrates that, even when an operator’s mathematical definition remains unchanged, achieving high performance requires experts to reassess bottlenecks and adapt implementations to the target hardware.
New demands driven by model evolution. Evolving model architectures not only introduce new operators but also change computational dependencies, state updates, and data-access patterns, making existing kernels difficult to reuse directly. For example, DeepSeek provides FlashMLA for multi-head latent attention (MLA) [9]. With the introduction of DeepSeek Sparse Attention (DSA), it also provides the corresponding sparse attention kernels through FlashMLA and implements the indexer’s scoring kernels in DeepGEMM [10 a]. As a linear-attention architecture, Gated DeltaNet (GDN) combines gating with the delta rule, and Kimi Delta Attention extends it with a finer-grained gating mechanism. Both create computational patterns with recurrent state updates [21, 13] that require specialized implementations distinct from standard attention. Operator implementations further branch into separate kernel paths for prefill and decode, forward and backward, and different precision and accumulation strategies. Sequence length and batch size further change the optimal design. Continued model evolution expands the kernel space along multiple dimensions, requiring sustained expert design and optimization for high-performance kernel development.
LLM-Based Kernel Generation
Existing LLM-based kernel generation methods generally build on a general-purpose agent and add specialized mechanisms for different challenges in kernel optimization: execution feedback and search organize iteration, independent validation ensures reliable evaluation, and domain knowledge and diagnostic tools help models locate performance bottlenecks.
Execution feedback and search. KernelBench evaluates models’ ability to generate faster, semantically equivalent kernels from PyTorch reference implementations and explores iterative refinement using execution and profiling feedback [18]. Existing work uses different search mechanisms to organize this optimization: METR uses parallel tree search [14], AI CUDA Engineer uses evolutionary search [19], K-Search guides search with a co-evolving world model [5], and AdaExplore distills recurring failures into reusable validity rules [11]. AVO introduces autonomous agents into the mutation step of evolutionary search, producing attention kernels that outperform expert implementations on some evaluated configurations [16].
Evaluation reliability. Optimization can exploit flaws in evaluation protocols. KernelBench-Verified reports that a GPT-5.5-generated ReLU kernel simply returns its input for known test shapes, yielding an apparent speedup [23]. To guard against such failures, kernel agents commonly incorporate independent validation, including sanitizer and anti-cheating checks in AKO4X [3] and independent acceptance checks by Codex in KDA (HAN Lab Kernel Mafia, 2026).
Domain knowledge and development environments. Timing reveals that a kernel is slow, but not why. Another line of work therefore equips agents with domain knowledge and diagnostic tools. CudaForge uses a Judge agent to interpret NCU metrics and guide revisions by a Coder agent [24]; KDA identifies the interface between models and the engineering loop as a primary bottleneck, motivating KernelWiki, built from production-code pull requests, an NCU profiling skill, and an IKET (in-kernel event tracing) profiling skill [15] (HAN Lab Kernel Mafia 2026); AKO4X maintains DSL-specific skills and a library of operator pitfalls [3]; and CAKE lets agents directly edit typed IR, with the compiler providing localized diagnostics before compilation [22].
General-purpose agent foundations and specialized layers. The mechanisms above are typically layered on the domain-independent tool use, context management, and longrunning autonomous execution provided by general-purpose agents such as Claude Code and Codex ([4] (OpenAI 2025)). KDA and AKO use Claude Code, while AVO uses an internal general-purpose agent, and add kernel-specific mechanisms for evaluation, knowledge, and workflow ([15, 1, 16]). Table 1 summarizes these mechanisms for the systems compared in § 4.
| System | Evaluation | Knowledge | Workflow |
| KDA | KDA-0.1 uses the FlashInfer-Bench standard; KDA-0.5 not released | KernelWiki (production PR corpus); NCU profiling skill; IKET (in-kernel event tracing) profiling skill | humanize2-based workflow; three-stage research–iterate–tune prompts; independent Codex acceptance checks at each step |
| AVO | Scoring function: correctness checks against a reference and throughput measurement | Knowledge base: CUDA/PTX documentation, Blackwell specifications, and FA4 source code | Single-lineage evolution; only non-regressing versions are committed; supervisor intervention upon stagnation |
| AKO | Compilation, correctness (including sanitizer), and timing benchmarks; anti-cheating checks | DSL-specific skills (including NCU profiling) and an operator-pitfall library (AKO4X) | Profile–modify–benchmark–commit loop with stagnation rules; multi-round campaigns with main and sub-agents (AKO4X) |
| CAKE | Pre-compilation validation of typed IR, cost-model ranking, localized diagnostics, and CUPTI timing | Recurring failures distilled into validation rules, IR primitives, and optimization tactics | Generate–filter–evaluate–route stages; single-shape evolution followed by library generalization |
| JustGoal | — | — | — |
Table 1. Kernel-specific mechanisms added to general-purpose agent foundations. KDA and AKO entries are based on their repositories and technical reports ([15, 12, 2 c]); AVO and CAKE entries are based on their papers ([16, 22]).
These approaches share a common pattern: each agent limitation motivates an additional mechanism. Yet these mechanisms were designed and validated with the underlying models available at the time; as those models improve, the marginal value of each mechanism must be measured again. JustGoal removes these kernel-specific layers and instantiates the goal-only baseline used for that measurement.
JustGoal: With Only a Goal
JustGoal
The box in Figure 1 contains the complete input received by the agent. Here, “goal-only baseline” denotes a control principle: give a general-purpose agent only the objective and constraints, without adding kernel-specific machinery. JustGoal instantiates this principle as the experimental configuration used in this paper. It retains only the general-purpose Claude Code agent substrate and access to the target hardware, while removing the entire kernel-specific layer listed in Table 1. The target hardware in all experiments in this paper is B200. Item 1 requires a clean environment precisely to ensure this.

Figure 1. The complete task input for JustGoal (English translation; the original Chinese prompt is given in Appendix A).
The model builds its own evaluation setup. The agent writes the compilation, correctness-checking, and timing code within its optimization loop: it first measures the reference kernel’s device time as the target to surpass, then writes a naive fp32 implementation as its correctness oracle.
This is the target operator we want: {operator config}. First measure its device time on a B200. Starting from scratch, and never looking at or copying the reference implementation, write a kernel that is at least 1.2× faster in device time.
Use an isolated, clean environment; do not use existing skills, memory, or other local files.
Make sure the kernel is correct. Preferably, first write a slow naive fp32 implementation and compare it with the reference; their difference gives a lower bound on the tolerance, which you multiply by a factor and then use to compare your implementation against the slow fp32 implementation.
Do not ask questions; run in
--dangerously-skip-permissionsmode.Use the B200 cluster for the GPU.
We explicitly note that the prompt contains one piece of expert guidance: item 2 instructs the agent to set the tolerance by multiplying the reference implementation’s error relative to the naive fp32 implementation by a factor. This is a common numerical-validation practice and does not involve kernel-optimization knowledge.
Evaluation Protocol
We separate kernel generation from final acceptance. The evaluation in this section is conducted independently by the authors after generation and determines whether code is included in the reported results.
Baseline: open-source library kernels. Candidates are limited to human-maintained library implementations: the official FlashAttention-4 kernels and FlashInfer (including FlashKDA). For each shape, we select the fastest candidate that passes correctness checks and is included in our evaluation.
Baseline: kernels generated by kernel agents. Candidates include kernels submitted by kernel agents to the MLSys’26 contest and publicly released kernels from KDA-0.5, AKO4X, CAKE, and AVO [15, 3, 22, 16]. The contest kernels are likewise produced by kernel agents, so they belong to this category rather than to the open-source libraries. For each shape, we likewise select the fastest candidate that passes correctness checks and is included in our evaluation. No publicly released specialized-agent kernel for FA4 backward is available for this comparison. Per-shape sources and selection details are provided in Appendix C.
Correctness verification. FA4 uses the official FlashAttention standard; every other operator uses the FlashInfer-Bench standard. Every implementation in the comparison must pass the corresponding standard.
Timing methodology. All kernels are timed by the authors with FlashInfer-Bench in a uniform environment. Before each measurement, the L2 cache is flushed (cold L2). CUPTI records the interval from the start of the first kernel launched by the entry point to the end of the last, excluding JIT compilation, cache flushing, and synchronization overhead.
Results
Experimental Configuration
The models are Anthropic’s Opus 4.8, Opus 5, and Fable 5.1, with reasoning effort high throughout; the agent environment is Claude Code 2.1.304. Kernel development and performance measurements run on NVIDIA B200 GPUs with driver 580.105.08 and CUDA 13.0.
Baselines and timing follow the protocol of § 3.2; agent-reported measurements are not used as results. The operators, shapes, and comparison groups of each experiment are given at the start of its subsection.
Overview: Comparison with Open-Source Libraries and Kernel Agents
This section gives an overview of JustGoal’s results on 11 operators and 42 shapes and answers two different questions: whether JustGoal’s kernels are faster than human-maintained open-source libraries, and whether they are faster than the kernels of specialized kernel agents. Each shape is one session, with Fable 5.1 as the primary model; a few sessions were completed by other models after a model fallback. We report geometric mean speedups per operator; per-shape latencies, the models actually used, and baseline sources are in Appendix § 10.
Against open-source libraries. The blue bars in Figure 2 are JustGoal’s results against open-source libraries. Excluding the DSA top-k indexer, whose baseline protocol remains unverified, 10 operators enter the comparison, and JustGoal is faster than the library on 8 of them, with geometric means from on FA4 backward to on GDN forward; it falls behind on FA4 forward () and Kimi Delta Attention ().

Figure 2. Per-operator geometric mean speedup of JustGoal (blue) and kernel agents (orange) over the open-source library; higher is better. The agent side takes the fastest contest or released-agent kernel per shape; parentheses name the sources. “—” indicates no comparable agent kernel. Per-shape details are in Appendix § 10.
Against kernel agents. The orange bars are the kernel agents’ kernels against the same open-source library, taking for each shape the fastest of the MLSys’26 contest kernel and the publicly released kernels of the four agent systems; the ratio of the blue bar to the orange bar is therefore JustGoal’s speedup over the agent kernel. Nine of the 10 plotted operators have a comparable agent kernel (FA4 backward has none); JustGoal is faster than the fastest agent kernel on five of them, ties on MoE FP8 (1.79× versus 1.80×), and falls behind on FA4 forward (0.77× versus AVO’s 1.18×), DSA forward (1.66× versus 1.72×), and Kimi Delta Attention (0.99× versus CAKE’s 1.27×). These agent kernels were produced by each system with its own choice of model and effort, so this comparison is between the kernels themselves; what the harness itself contributes under the same model is answered in § 4.3 and § 4.4.
Relation to the released KDA-0.5 kernels. On the nine DSA, GDN, and MoE shapes, the kernels released in the KDA-0.5 repository reach a geometric mean of 1.29×, and single-shape JustGoal reaches 1.20× on the same nine shapes, close to the release; per-shape values are in Appendix § 10. Our reproduction of its harness with Fable 5.1 reaches 0.91× (same-GPU paired protocol, Table 7 in the appendix); the model and effort behind the release are not public, and this gap bounds the fidelity of the reproduction.
| Operator | Shape | Reference kernel | Opus 4.8 JustGoal | Opus 4.8 KDA-0.5 | Opus 5 JustGoal | Opus 5 KDA-0.5 | Fable 5.1 JustGoal | Fable 5.1 KDA-0.5 |
| FA4 backward | MHA | FA4 | ||||||
| GQA | FA4 | |||||||
| varlen MHA | FA4 | |||||||
| DSA forward | heavy load | contest kernel | ||||||
| small request | contest kernel | |||||||
| mixed lengths | contest kernel | |||||||
| GDN forward | long sequence | contest kernel | ||||||
| single short seq. | contest kernel | |||||||
| few long seqs. | contest kernel | |||||||
| MoE FP8 | single token | contest kernel | ||||||
| medium batch | contest kernel | |||||||
| large prefill | contest kernel | |||||||
| single-shape JustGoal / KDA-0.5 repro., geomean | (12 shapes) | (12 shapes) | (12 shapes) |
Table 7. Per-shape speedup in the same-model comparison (higher is better). FA4 backward uses FA4 as the reference; the other operators use the MLSys’26 contest kernel. The faster configuration per model is bold. The scope-matched comparison is in Table 9.
The comparison with human-written libraries needs no further argument: JustGoal is faster on 8 of the 10 operators, and faster than the fastest agent release on 5 of the 9 operators that have one. Whether specialized machinery adds value beyond this result is a separate question. The next section first examines how the gap between full configurations changes across models; the following scope-matched comparison directly measures the mechanism gain.
Configuration Gaps Across Models
Configuration. Four operators (FA4 backward, DSA forward, GDN forward, MoE FP8), three shapes each, 12 shapes in all; every run is repeated with each of the three models under the same 24-hour budget. This section and Section 4.4 compare four configurations:
JustGoal, single shape: one session optimizes one shape. 12 shapes × 3 models, 36 sessions.
JustGoal, whole definition: the same prompt, with the goal changed to every workload of the operator’s FlashInfer-Bench definition (23 for DSA, 100 for GDN, 19 for MoE FP8, 3 for FA4 backward); the session may dispatch by shape, and the final solution must pass the official evaluator on every workload. 4 operators × 3 models, 12 sessions.
KDA-0.5 reproduction: the whole-definition scope, run through the KDA-0.5 harness with the same three models. KDA-0.5 has not fully released its prompts, so we follow its repository and technical report ([15] (HAN Lab Kernel Mafia 2026)): the humanized
flame_chasetwo-chaser workflow, each turn a fresh session sharing the git workspace; the KernelWiki, NCU report parsing, and iket-profiling skills; the public KDA 0.1 contest prompt as the task and FlashInfer-Bench as the evaluator. 4 operators × 3 models, 12 runs.KDA-0.5 reproduction, single shape: the same KDA-0.5 harness; as in JustGoal, single shape, each run optimizes only one shape. Run with Fable 5.1 only: 12 shapes, 12 runs.
The reference kernel is the FA4 library kernel for FA4 backward and the MLSys’26 contest kernel for the other three operators. Every implementation follows the retest of the public repository. Two runs violated the task constraints and are listed as is: the Opus 4.8 JustGoal run on DSA heavy load read the reference implementation, and the Opus 4.8 KDA-0.5 reproduction on MoE called FlashInfer library kernels. The kernels released in the KDA-0.5 repository were produced with an unknown model and are therefore not part of the same-model comparison; their comparison with JustGoal is in Section 4.2.
Results. The first three configurations exist for all three models. Figure 3 gives, per operator, the geometric mean speedup of single-shape JustGoal and of the KDA-0.5 reproduction over the reference kernel across models; per-shape values are in Table 7 in the appendix. The two differ in task scope, so this section only examines the configuration gap; the scope-matched comparison is in § 4.4.

Figure 3. Per-operator geometric mean speedup of single-shape JustGoal and the KDA-0.5 reproduction (whole definition) over the reference kernel, by model; the dashed line is the reference kernel.
The gap between configurations changes across models. Over Table 7, single-shape JustGoal’s geometric mean speedup over the reference kernel rises from with Opus 4.8 to with Opus 5 and with Fable 5.1 (12 shapes each), and the same-model KDA-0.5 reproduction from to and : from Opus 4.8 to Fable 5.1, JustGoal improves by about and the KDA-0.5 reproduction by about . JustGoal trails the same-model KDA-0.5 reproduction with Opus 4.8 and Opus 5 and overtakes it at Fable 5.1, where its geometric mean exceeds on all four operators while the KDA-0.5 reproduction stays below on GDN. The two configurations use different task scopes, so this comparison only shows that their relative performance changes with the model; it cannot attribute the gap to any mechanism in the harness.
Harness and Task Scope: The Fable 5.1
To distinguish the influence of harness configuration from that of task scope, we construct a comparison with Fable 5.1: {JustGoal, KDA-0.5 harness} {single shape, full workload coverage}. Full workload coverage requires one run to cover every workload in the operator’s FlashInfer-Bench definition, allowing per-shape dispatch or specialization. Table 2 reports per-operator geometric mean speedups for the four configurations; per-shape results are in Table 6 in the appendix.
| Operator | single shape | full workload coverage | ||
| JustGoal | KDA-0.5 | JustGoal | KDA-0.5 | |
| FA4 backward | ||||
| DSA forward | ||||
| GDN forward | ||||
| MoE FP8 | ||||
| Geomean |
Table 2. Harness and task scope with Fable 5.1: a comparison. Columns report per-operator geometric mean speedups over the reference kernels; the final row aggregates all 12 shapes.
| Operator | Shape | JustGoal, single shape | KDA-0.5, single shape | JustGoal, whole definition | KDA-0.5, whole definition |
| FA4 backward | MHA | ||||
| GQA | |||||
| varlen MHA | |||||
| DSA forward | heavy load | ||||
| small request | |||||
| mixed lengths | |||||
| GDN forward | long sequence | ||||
| single short seq. | |||||
| few long seqs. | |||||
| MoE FP8 | single token | ||||
| medium batch | |||||
| large prefill | |||||
| Geomean |
Table 6. Per-shape speedup of the four configurations over the reference kernel with Fable 5.1 (higher is better). The fastest per row is bold; per-operator geometric means are in Table 2.
At single shape, JustGoal and the KDA-0.5 harness achieve geometric mean speedups of and over the reference kernels, respectively, so the specialized harness provides no overall improvement in this setting. With full workload coverage, the corresponding speedups are and , giving the KDA-0.5 harness a advantage over JustGoal. The harness’s relative advantage thus emerges as task scope broadens, although both configurations achieve lower overall performance than in their respective single-shape settings.
On the tested tasks, these results show that Fable 5.1 can already achieve good performance when optimizing a single shape, but covering all shapes in one run does not yet match the overall performance of independent per-shape optimization. One possible explanation is that the model still struggles to cluster shapes by their computational characteristics and decompose the optimization task into subproblems that can be addressed separately. This explanation requires further experimental validation: the present comparison neither establishes the specific cause of the performance decline nor attributes it to any individual harness mechanism. These observations are limited to Fable 5.1; the other two models were not evaluated under all four configurations, and their results are reported in Table 8 in the appendix.
| Operator | Model | JustGoal, single shape | JustGoal, whole definition | KDA-0.5 reproduction |
| FA4 backward | Opus 4.8 | |||
| Opus 5 | ||||
| Fable 5.1 | ||||
| DSA forward | Opus 4.8 | |||
| Opus 5 | ||||
| Fable 5.1 | ||||
| GDN forward | Opus 4.8 | |||
| Opus 5 | ||||
| Fable 5.1 | ||||
| MoE FP8 | Opus 4.8 | |||
| Opus 5 | ||||
| Fable 5.1 |
Table 8. Per-operator geometric mean speedup of the three configurations over the reference kernel (higher is better). Single-shape values are computed from Table 7; per-workload latencies are in Appendix C.2.
Ablation Study
Fable 5.1 on three shapes, FA4 backward varlen, DSA forward heavy load, and GDN forward long sequences, with default JustGoal as the reference and one mechanism added at a time: a link to the KernelWiki repository in the prompt (Kernel Wiki); an instruction to use NCU for performance analysis (NCU); a report-parsing skill on top of the NCU instruction (NCU skill); and the goal prefix replaced by the humanize2 flame_chase two-agent workflow (flame chase).
| Shape | JustGoal | Kernel Wiki | NCU | NCU skill | flame chase |
| FA4 backward varlen | |||||
| DSA forward heavy | |||||
| GDN forward long |
Table 3. Speedup over the reference kernel under each configuration (higher is better). The best per row is bold; marks a result faster/slower than JustGoal.
These ablations yield two observations: apart from NCU skill, the KDA-0.5 designs examined here show no consistent benefit, suggesting that the benefits of specialized harnesses need to be reassessed with stronger models; NCU itself works well without a dedicated skill, and adding one provides no further improvement. This suggests that, with stronger models, the value lies in letting the model know which tools are available; specialized mechanisms built around those tools do not necessarily bring additional gains.
Accuracy verification. This comparison contrasts two kinds of accuracy constraint. Without a hard constraint, JustGoal only asks for accuracy verification in the prompt and says how to set the tolerance; the model writes the check itself, and nothing outside it stops a kernel that misses the tolerance from becoming the final version. With a hard constraint, as in the KDA-0.5 reproduction, the harness runs the official evaluator in the loop and a failing kernel cannot be submitted. Under the hard constraint the evaluator decides: all 12 reproductions with Opus 5 and Fable 5.1 pass the retest, and the 3 Opus 4.8 DSA kernels fail for not returning lse, a case the evaluator did not catch at the time. Without it, Table 4 counts, per JustGoal configuration, the directories whose released version passes the retest of the public repository: the model can build its own numerical check, yet some kernels still fail accuracy. Single-shape JustGoal passes 11/12, 9/12 and 12/12 on the four main operators, 1/3, 4/7 and 22/26 on the other seven, and the whole-definition runs 5/5, 4/5 and 5/5. The failures are accuracy, missing outputs, build or runtime errors. With the stronger model the model-built check approaches the pass rate of the hard constraint but does not replace it.
| Configuration | Opus 4.8 | Opus 5 | Fable 5.1 |
| JustGoal single shape, 4 main operators | 11/12 | 9/12 | 12/12 |
| JustGoal single shape, other 7 operators | 1/3 | 4/7 | 22/26 |
| JustGoal whole definition | 5/5 | 4/5 | 5/5 |
Table 4. Implementation directories whose released version passes the independent retest / directories retested, per JustGoal configuration; all generation-time correctness checks were built by the model.
Optimization Case Study
We use GDN (Gated DeltaNet) forward on long sequences (8192 tokens in 20 sequences of length 15 to 3371) to show how JustGoal crosses the contest kernel from scratch. The reference is the MLSys’26 contest kernel from the KDA-0.5 submission, which the agent could only call as a black box; the audited trajectory never reads its source. Fable 5.1 ran 8.9 hours without intervention. Latencies are the session’s own L2-flushed kernel span; the reference is 125.1 s. Appendix § 9 gives the full record and a second case, FA4 backward GQA.
The trajectory (Figure 4) has three stages, each ending when the agent measured that its toolchain had run out of headroom. Triton started at 302 s (0.41×) and the agent put its floor near 146 s. After microbenchmarking mma.sync on B200, it hand-wrote CUDA kernels and, guided by NCU stall counters and in-kernel cycle counts, reached 165 s at 1.3 hours; further overlap and cluster work only reached 153.8 s. At 4.2 hours it judged the path near the shared-memory bandwidth limit and moved the recurrence to B200’s tcgen05 tensor cores, checking descriptor encodings and TMEM layouts with probe kernels. The first tcgen05 version was slower (235 s); warp specialization, fusing the transposed O product into the recurrence’s MMA group, and running the intra-chunk kernel on the remaining SMs crossed the reference at 6.2 hours. The final version measures 95.6–97.3 s over three processes (1.29–1.31×; 1.32× under the paired re-measurement), with errors within 1.30 times the reference’s.

Figure 4. Optimization trajectory of JustGoal (Fable 5.1) on GDN forward, long sequences. The staircase is the session’s best self-measured value; gray points are log-reviewed change batches that did not improve it, and shading marks strategy phases. Session measurements and the main table use different protocols.
Profiling, toolchain switching and hardware probing were all the agent’s own choices. The prompt names no profiler, supplies no domain corpus and prescribes no stages, whereas KDA-0.5 encodes the same behaviors as NCU and in-kernel tracing skills, KernelWiki and three-stage prompts (Table 1). Given only the goal, the model carried out the expert workflow on its own.
Discussion
The value boundary of a specialized harness changes with foundation-model capability and task scope. In the equal-scope Fable 5.1 comparison, the full harness produces no overall gain on single shapes, and single-mechanism ablations show no consistent benefit from domain knowledge or workflows. At whole-definition scope, however, the harness still helps the model advance more workloads simultaneously. This suggests that the primary value of specialized machinery may be shifting from raising the peak performance of a single target to helping models organize a broader task scope. Task scope is also critical. The performance decline under full workload coverage shows that making one implementation cover all workloads of an operator is a generality tax, and current models cannot yet partition the workloads well enough on their own to avoid it. This leaves two paths. One is to make the agent’s own partitioning ability a design goal of kernel agents and assess progress through the performance difference between single-shape and full-workload configurations in Table 2. The other is deep customization: in high-value scenarios with fixed configurations, no single implementation has to cover every workload, the generality tax disappears entirely, and the single-shape column of Table 6 is the level reached once it does. We call the latter Forge Engineering. Each implementation optimizes performance only for its target configuration but must remain correct for any input of that configuration; the costs are optimization sessions that grow with the number of configurations, plus per-configuration verification.
JustGoal surpasses expert implementations on some tasks studied in this paper, suggesting that, as foundation models improve, the relative cost-effectiveness of having experts hand-write every kernel from scratch is declining. Meanwhile, the gain from a specialized harness varies with the model and task scope, showing that expert mechanisms must be matched to the actual capabilities of the foundation model. We believe a more promising path is to combine the depth of expert knowledge with the model’s broad prior knowledge, autonomous experimentation, and scalable search to approach hardware performance limits. Concretely, JustGoal can serve as a starting point that is updated with the model: let the model explore independently first, then have experts analyze optimization trajectories, profiling evidence, and failure cases to identify specific limitations in bottleneck diagnosis, hardware knowledge, search organization, or workload coverage, and add knowledge, diagnostic tools, or constraints accordingly. The value of each added mechanism should still be measured by its gain over JustGoal using the same model.
Limitations
Our experiments focus on selected operators and shapes on NVIDIA B200 GPUs; the applicability of our findings to other hardware and tasks remains to be validated. Comparisons with publicly released kernels concern the kernels themselves; the systems use different models and optimization budgets. The same-model comparisons in § 4.3 separately examine the harness. KDA has released its general workflow, base prompts, and related skills, but the KDA-0.5 results repository primarily provides the final kernels and retest code rather than the complete prompts, workflow configuration, and search process used for that version. Our reproduction can therefore only follow the available public materials and may differ from the original system. KDA-0.5 also bundles domain priors with scope-organization mechanisms, and we do not ablate these two groups separately. Their distinction in § 4.4 is our interpretation of the data.
Our correctness validation remains limited by test coverage. FA4 uses the official FlashAttention standard, and every other kernel uses the FlashInfer-Bench standard. Passing error-tolerance checks on a finite set of test inputs does not guarantee correctness for all valid inputs and execution conditions. The reported performance results should therefore be understood as results obtained after passing the adopted validation procedures, rather than a comprehensive guarantee of production readiness.
Conclusion
We first identify a change in current model capability. Given only objectives and constraints, without predefined kernel-specific evaluation infrastructure, domain knowledge, or search workflows, a general-purpose agent can construct its own evaluation pipeline and iteratively optimize kernels. On NVIDIA B200 GPUs, JustGoal is faster than human-maintained open-source libraries on 8 of 10 operators and faster than the fastest specialized-agent release on 5 of 9. Without task-time injection of kernel-specific knowledge or optimization workflows, current frontier models can establish evaluation feedback, identify performance bottlenecks, adapt optimization strategies, and cross the performance line represented by human-maintained implementations on a range of operators.
We further propose an evaluation principle that does not depend on the current level of model capability. JustGoal is not a new agent architecture, but the experimental configuration used to observe the performance level of a goal-only setup and measure the marginal value of explicit expert mechanisms. Specialized machinery should be considered effective only when it clearly outperforms this baseline. When model capabilities change, the returns of existing mechanisms should be measured again rather than inherited from earlier results. The current model still benefits from a harness at broader task scope; whether future models do so should also be answered by a scope-matched comparison. We hope future work will not only apply new models to existing systems but also use a goal-only configuration under the same model and task scope as the baseline, reexamine each component’s contribution against it, and determine which components to retain, adapt, or remove.
Original Chinese Prompt
Figure 1 is an English translation; the box below is the original Chinese prompt given to the agent.
这是我们要的目标算子(算子 config),请先在 B200 上测量它的 device time 时间。从 0 开始,绝对不看或抄写这个 ref 的代码实现,写出一个至少达到他 1.2X 倍快 device time 速度的 kernel 实现。
请使用一个独立的干净的环境,不要用已有的 skills,memory,或者其它本机文件。
请务必确保正确性。最好通过先写一个 fp32 的 naive 慢实现,再和 ref 比他们两的差这样可以获得一个下限容差范围再乘个值,然后用这个容差来对比你的实现和 fp32 慢实现的。
不要问问题,用 --dangerously-skip-permissions 模式。
GPU 请使用 B200 集群。
Optimization Trajectories in Detail
This appendix gives the full record behind § 4.6: a second case, FA4 backward GQA, and the step-by-step trajectory of the GDN forward case summarized there. Both have complete trajectories, and all per-version latencies are the sessions’ own measurements.
FA4 Backward GQA
We use FA4 backward GQA (, , , , , causal, bf16). The reference is FlashAttention-4 by Tri Dao et al. (Dao et al. [7]), which the agent can invoke only as a black box through its public API. We audited the complete trajectory and confirmed that the agent did not read its source code. Fable 5.1 chose Triton Gluon to implement the kernel and completed the task in four hours without human intervention.
The agent first measured the reference kernel’s device time and wrote a naive fp32 implementation as a correctness oracle. Its first kernel took 2771 s and passed correctness checks on the first attempt, but was approximately 80% slower than the reference. The agent then profiled stall reasons with NCU, added ablation switches and %clock instrumentation to measure cycles per stage. It found that MMA alone required only about 830 s, while softmax and atomic accumulation of dQ accounted for most of the time. By reusing TMEM and switching dQ accumulation to TMA bulk reduction, it reduced latency to approximately 1320 s after 2.5 hours.
Over the next 1.4 hours, more than a dozen new versions improved the best latency only to 1278 s (Figure 5). The agent wrote microbenchmarks and measured TMA reduction throughput of only 25–30 B/clk per SM, with roughly 2800 cycles required to release a staging buffer. Disabling dQ reduction reduced the main kernel’s latency to 931 s. These observations pointed to the same cause: reduction reused the staging buffer for dO, blocking dO loads. The agent therefore allocated a dedicated 32 KB staging buffer for reduction and moved dO loads to drain warps using cp.async, leaving TMA exclusively for reduction. This change reduced latency to 1166 s in a single step. Under the unified cold-L2 paired re-measurement, the final kernel runs in 1321 s against 1676 s for FA4, a speedup of 1.27×. The final kernel accumulates dQ in FP16 and rounds dS to BF16, so its internal numerics differ from FA4; its dQ, dK and dV all pass the gradient tolerance of the official FlashAttention standard, with errors at 20–33% of the allowed budget.

Figure 5. Optimization trajectory of JustGoal (Fable 5.1) on FA4 backward GQA. The figure uses session measurements; the final independent result is in the main table.
Throughout this process, the agent independently used NCU, ablations, and hardware microbenchmarks to diagnose bottlenecks and adjusted the kernel structure accordingly, ultimately producing an operator that outperforms the expert implementation.
GDN Forward, Long Sequences
The workload, reference and measurement protocol are those of Section 4.6; this is the full record of the 8.9-hour run.
The agent first wrote a Triton version that passed correctness at once but took 302 s, 0.41 the reference (Figure 4). Sweeping tile and warp configurations showed a fixed per-step overhead in the recurrence kernel and register spills in the intra-chunk kernel, with a floor of about 146 s for the Triton path, so it abandoned Triton, first wrote microbenchmarks for mma.sync throughput and dependency latency on B200, and then hand-wrote two CUDA C++ kernels with mma.sync and ldmatrix, reaching 194.5 s at 0.5 hours. It then checked spills with cuobjdump, located hot spots with NCU stall and source counters, and counted cycles per phase inside the kernel, bringing the recurrence from 4025 to 1993 cycles per step and the total to 165 s at 1.3 hours. The next two hours gained little: overlapping the three kernels on two streams with per-block flags made things slower; several launch failures and deadlocks were debugged with compute-sanitizer; thread-block clusters with TMA multicast let 8 CTAs share one broadcast of the W/K tiles; and globaltimer timelines of every CTA’s work items showed a busy ratio of only 0.74 with about 5 s gaps between items, so it switched to static round-robin assignment to remove the atomics. At 3.8 hours it stood at 153.8 s, 0.81.
At 4.2 hours the agent reported plainly that correctness was met but performance was not, gave a breakdown of where the time went, and judged that the mma.sync path was near the shared-memory bandwidth limit, so the recurrence had to move to B200’s fifth-generation tensor cores (tcgen05, whose operands are read from shared memory by the tensor core and whose accumulators live in TMEM). It consulted the PTX ISA documentation and CUTLASS headers, wrote probe kernels to verify descriptor encodings, swizzles and TMEM layouts, and rewrote the recurrence kernel. The first tcgen05 version took 235 s, about 3 s per step, bottlenecked by 24 serial TMEM waits. Over the next three hours it dumped TMEM intermediates and rebuilt O in fp32 to localize a numerical error, fused the transposed O product into the recurrence’s MMA group, microbenchmarked A operands from TMEM versus shared memory, split TMA production, MMA issue and output write-back into dedicated warps, and launched the recurrence first at the highest stream priority with the intra-chunk kernel running on the remaining SMs. It first crossed the reference at 6.2 hours (122 s) and reached 101 s at 7.5 hours. Finally it made the intra-chunk kernel process one head per CTA with the longest sequences first, reaching 94.6 µs at 8.5 hours; the final version measured 95.6 to 97.3 µs over three independent processes, 1.29–1.31× the reference, with output and state errors at 1.00 and 1.30 times the reference error. Under the same-GPU paired re-measurement it reaches 1.32× over the MLSys’26 contest kernel shipped with KDA-0.5.
Per-Shape Latencies and Baseline Sources
Table 5 lists implementation latencies and baseline sources. The library column only lists human-maintained library implementations; the MLSys’26 contest kernels are listed separately and, together with the released agent kernels, form the kernel-agent baseline of § 4.2, taking the faster of the two per shape. Some experiments were completed with other models due to model fallback: the medium- and large-batch DSA top-k indexer tasks used Opus 5.5, while the short and medium MLA paged prefill tasks and the medium- and large-batch GQA paged decode tasks used Opus 5. The target model for all other tasks was Fable 5.1. The original length distributions of four shapes were not recovered, so they were measured on the substitute distribution of an earlier round: GQA paged decode medium and large batch, Kimi Delta Attention packed mixed lengths, and MLA paged prefill large prefill.
The three DSA top-k indexer candidates passed correctness retests on their respective limited inputs, with cold-L2 kernel spans of 47.59, 3531.5, and 13.25 µs; the large-batch result uses signed weights. The three historical top-k baselines (AKO4X fast_split_v6 and v8, and Dogacel) compress scores to BF16 before selection and fail a strict FP32 membership check on 9 of their 18 input configurations. The agent kernel column gives the AKO4X result in the retest repository, as a range over non-negative and signed weights; Dogacel measures 1.44–1.47, 15.24–24.46 and 18.02–27.43 µs. The library and contest columns remain historical values. None of these baselines was paired with the candidates on the same GPU, so the latencies are retained as audit records and are not used to compute speedups in Figure 2.
Absolute latencies from different measurement batches are not comparable; every speedup uses the baseline of its own pair. On GDN long sequences the contest kernel measured 209.5 µs in the batch paired with JustGoal and 124.9 µs in the independent baseline retest; across the batches of this appendix, the FA4 backward baseline spans 1371–1482, 1456–1676 and 1572–1706 µs on MHA, GQA and varlen MHA.
| Operator | Shape | JustGoal | Library | Contest kernel | Released agent kernel |
| FA4 backward | MHA | 1357.9 | 1481.9 (FA4) | — | — |
| FA4 backward | GQA | 1321.2 | 1675.5 (FA4) | — | — |
| FA4 backward | Variable-length MHA | 1581.9 | 1705.6 (FA4) | — | — |
| FA4 forward | MHA (8,4096) non-causal | 874.0 | 981 (FA4) | — | 699 (AVO) |
| FA4 forward | MHA (1,32768) non-causal | 6709.6 | 5930 (FA4) | — | 5270 (AVO) |
| FA4 forward | MHA (8,4096) causal | 544.0 | 460 (FA4) | — | 395 (AVO) |
| FA4 forward | MHA (1,32768) causal | 4188.9 | 3170 (FA4) | — | 2690 (AVO) |
| FA4 forward | GQA g8 (8,4096) non-causal | 2277.2 | 1558 (FA4) | — | 1336 (AVO) |
| FA4 forward | GQA g8 (1,32768) non-causal | 13205.8 | — | — | — |
| FA4 forward | GQA g8 (8,4096) causal | 2501.4 | 960 (FA4) | — | 749 (AVO) |
| FA4 forward | GQA g8 (1,32768) causal | 6667.6 | 5750 (FA4) | — | 5410 (AVO) |
| FA4 forward | GQA g4 (8,4096) non-causal | 1767.0 | 1406 (FA4) | — | 1347 (AVO) |
| FA4 forward | GQA g4 (1,32768) non-causal | 15886.9 | — | — | — |
| FA4 forward | GQA g4 (8,4096) causal | 1003.0 | 929.5 (FA4) | — | 751 (AVO) |
| FA4 forward | GQA g4 (1,32768) causal | 8321.2 | 6030 (FA4) | — | 5340 (AVO) |
| Operator | Shape | JustGoal | Library | Contest kernel | Released agent kernel |
| DSA forward | Heavy workload | 11.42 | 13.71 (FlashInfer) | 10.27 (MLSys’26) | 10 (KDA-0.5) |
| Small request | 2.891 | 7.03 (FlashInfer) | 4.037 (MLSys’26) | 3 (KDA-0.5) | |
| Mixed lengths | 7.115 | 11.07 (FlashInfer) | 8.427 (MLSys’26) | 7 (KDA-0.5) | |
| DSA top-k indexer | Single request, short sequence | 47.59 | 154.88 (DeepGEMM + FlashInfer) | 2.35 (MLSys’26) | 1.47–1.55 (AKO4X, FP32 fix) |
| Medium batch | 3531.5 | 315.59 (DeepGEMM + FlashInfer) | 17.21 (MLSys’26) | 17.39–17.70 (AKO4X, FP32 fix) | |
| Large batch, mixed lengths | 13.25 | 313.28 (DeepGEMM + FlashInfer) | 21.73 (MLSys’26) | 24.24–24.49 (AKO4X, FP32 fix) | |
| GDN forward | Long sequences | 158.5 | 145.6 (FlashInfer) | 209.5 (MLSys’26) | 126 (KDA-0.5) |
| Single short sequence | 4.133 | 45.5 (FlashInfer) | 5.936 (MLSys’26) | 6 (KDA-0.5) | |
| Few long sequences | 105.5 | 133 (FlashInfer) | 131.6 (MLSys’26) | 105 (KDA-0.5) | |
| GDN decode | Single request | 2.347 | 3.39 (FlashInfer) | 2.35 (MLSys’26) | 2.31 (AKO4X) |
| Medium batch | 4.725 | 5.56 (FlashInfer) | 5.27 (MLSys’26) | 4.75 (AKO4X) | |
| Large batch | 11.91 | 13.66 (FlashInfer) | 12.51 (MLSys’26) | 12.79 (AKO4X) | |
| Kimi Delta Attention | Packed mixed lengths | 740.0 | 455.6 (FlashKDA) | — | 391 (CAKE) |
| Single fixed-length long sequence | 454.9 | 658.7 (FlashKDA) | — | 470 (CAKE) | |
| Packed equal lengths | 506.7 | 545.1 (FlashKDA) | — | 435 (CAKE) | |
| MLA paged decode | Single request | 6.320 | 15.98 (FlashInfer) | — | 9.38 (AKO4X) |
| Medium batch, mixed lengths | 15.57 | 31.25 (FlashInfer) | — | 17.36 (AKO4X) | |
| Large batch | 57.58 | 77.5 (FlashInfer) | — | 69.97 (AKO4X) | |
| MLA paged prefill | Short prefill | 4.981 | 22.886 (FA4) | — | 10.04 (AKO4X) |
| Medium prefill | 23.65 | 34.73 (FA4) | — | 22.29 (AKO4X) | |
| Large prefill, mixed lengths | 430.6 | 553.0 (FA4) | — | 2491 (AKO4X) | |
| GQA paged decode | Single request | 4.853 | 7.488 (FA4) | — | 7.74 (AKO4X) |
| Medium batch, short sequences | 10.15 | 16.07 (FlashInfer) | — | 11.55 (AKO4X) | |
| Large batch | 60.22 | 69.61 (FA4) | — | 63.47 (AKO4X) | |
| MoE FP8 | Single token | 50.09 | 86.41 (FlashInfer) | 57.15 (MLSys’26) | 49.05 (KDA-0.5) |
| Medium batch | 296.9 | 377.41 (FlashInfer) | 302.3 (MLSys’26) | 278.29 (KDA-0.5) | |
| Large prefill | 967.1 | 2544 (FlashInfer) | 1173.6 (MLSys’26) | 1042.65 (KDA-0.5) |
Table 5. Per-shape latency (µs, lower is better) and baseline sources. The library column contains only human-maintained implementations; contest and released-agent kernels together form the kernel-agent baseline in the main text.
caption:
:::
Full Configuration Results for All Three Models
Table 6 gives the per-shape speedups of the four Fable 5.1 configurations of § 4.4; Table 2 in the main text is its per-operator geometric mean. Table 7 gives the per-shape speedups of three configurations under all three models; its Fable 5.1 JustGoal column equals that of Table 6. Table 8 gives the operator-level geometric means of the three configurations under all three models; the two effects of Table 9 are ratios of it.
| Operator | Opus 4.8 | Opus 5 | Fable 5.1 | ||||||
| scope | harness | vs. single | scope | harness | vs. single | scope | harness | vs. single | |
| FA4 backward | |||||||||
| DSA forward | |||||||||
| GDN forward | |||||||||
| MoE FP8 | |||||||||
| Geomean |
Table 9. Scope effect, harness effect, and gap to the single-shape ceiling per operator under the three models; last row: geometric mean.
Table 9 reduces the results of the three models to three ratios, each dividing the operator’s geometric means over three shapes: scope effect = whole definition ÷ single shape, below 1 is the loss from widening the goal; harness effect = KDA-0.5 reproduction ÷ whole definition, above 1 means that the KDA-0.5 configuration is faster at equal scope; vs. single = KDA-0.5 reproduction ÷ single-shape JustGoal. In 11 of the 12 operator-model combinations, whole-definition JustGoal trails the same-model KDA-0.5 reproduction, and it crosses the reference kernel in none. The harness and scope effects are and with Opus 4.8, and and with Fable 5.1. These ratios describe gaps between full configurations and do not isolate the contribution of any individual mechanism.
Per-Workload Latencies of the Definition-Level Comparison
Table 10 lists the per-workload latencies behind Table 9. All values were re-measured by the authors on one B200 with the official evaluator (flashinfer-bench main @ 40e6ca7: CUPTI kernel span, cold L2, median, each definition’s official tolerances) in at least two alternating rounds; the whole-definition solutions pass the official evaluator on every workload of their definitions, and only the three shapes used in the main text are timed here. The KDA-0.5 reproduction is the version at the end of the run. Library column: FA4 4.0.0b19 for FA4 backward; for DSA, the FlashInfer baseline solution of the contest dataset, which concatenates its inputs on every call and does not reflect the library kernel’s own speed; for MoE, FlashInfer trtllm_fp8_block_scale_moe. The MoE Opus 5 whole-definition solution is bimodal across rounds on the single-token workload (63–122 s); the median is shown. The sessions’ own reports are generally more favorable than these numbers: the GDN Opus 4.8 session read a millisecond timer as microseconds, the MoE Opus 4.8 session’s benchmark hit a CUDA graph cached on input pointers, and the MoE Opus 5 session timed itself on inputs it generated. These rows await the same-GPU paired re-measurement of § 3.2.
| Operator | Shape | JustGoal, whole definition | KDA-0.5 reproduction | Contest kernel | Library |
| FA4 backward | MHA | 2675 / 1702 / 1932 | 1337 / 1416 / 1200 | — | 1371 |
| FA4 backward | GQA | 2701 / 1647 / 1565 | 1238 / 1478 / 1163 | — | 1456 |
| FA4 backward | varlen MHA | 3426 / 1985 / 2110 | 1557 / 1697 / 1378 | — | 1572 |
| DSA forward | heavy load | 19.3 / 9.58 / 13.3 | 14.1 / 8.30 / 7.43 | 10.13 | 585 |
| DSA forward | small request | 10.0 / 4.02 / 4.73 | 5.93 / 4.41 / 3.56 | 3.36 | 661 |
| DSA forward | mixed lengths | 13.5 / 6.90 / 9.30 | 9.22 / 6.29 / 5.11 | 6.92 | 652 |
| GDN forward | long sequence | 1126 / 213 / 178 | 710 / 126 / 234 | 124.9 | — |
| GDN forward | single short seq. | 13.96 / 5.96 / 5.36 | 15.2 / 4.74 / 4.67 | 4.81 | — |
| GDN forward | few long seqs. | 1863 / 198 / 203 | 973 / 138 / 312 | 107.7 | — |
| MoE FP8 | single token | 294 / 300 / 53.7 | 135 / 72.9 / 42.1 | 52.0 | 149 |
| MoE FP8 | medium batch | 551 / 433 / 431 | 336 / 386 / 280 | 275.8 | 435 |
| MoE FP8 | large prefill | 2703 / 1390 / 2431 | 1271 / 1582 / 923 | 1028.5 | 2515 |
Table 10. Per-workload device time behind Table 9 (, lower is better). Each cell is Opus 4.8 / Opus 5 / Fable 5.1.
Correctness was re-checked under the protocol of § 3.2: FA4 uses the official FlashAttention standard, and every other operator uses the FlashInfer-Bench standard. The whole-definition solutions of all twelve operator-model combinations (fifteen directories, since FA4 backward has a fixed-length and a varlen definition) pass on every timed workload, with three exceptions: the Opus 4.8 and Opus 5 DSA solutions exceed the lse tolerance by – on the small-request workload, where KDA-0.5’s lse is almost exact and the tolerance is tiny, both absolute errors being below ; and the Opus 4.8 MoE solution gets 3 rows wrong as a whole on large prefill, apparently near-ties in expert selection, with the mean error equal to KDA-0.5’s, which the official evaluator passes through its 90% matched-ratio rule. In the KDA-0.5 reproduction, the Opus 4.8 DSA solution returns only the attention output and no lse, which the official evaluator did not catch; the Opus 4.8 MoE solution calls FlashInfer library kernels and fails the standard above on two further workloads. Two items failed the public-repository retest as released: the three Opus 4.8 DSA KDA-0.5 reproduction implementations matched an earlier failing upstream version, and the Opus 5 MoE whole-definition solution hit a Triton launch-parameter mismatch on all 19 workloads (36 parameters expected, 27 given).
References
- [1]AKO Team. AKO: Agentic kernel optimization. https://github.com/TongmingLAIC/AKO, 2026a.
- [2]AKO Team. AKO4ALL: Agentic kernel optimization for all. https://github.com/TongmingLAIC/AKO4ALL, 2026b.
- [3]AKO Team. AKO4X: Agentic kernel optimization, advanced and extensible. https://github.com/TongmingLAIC/AKO4X, 2026c.
- [4]Anthropic. Claude code. https://claude.com/claude-code, 2025.
- [5]Shiyi Cao, Ziming Mao, Joseph E. Gonzalez, and Ion Stoica. K-Search: LLM kernel generation via co-evolving intrinsic world model, 2026. arXiv:2602.19128.arxiv.org/abs/2602.19128
- [6]Tri Dao, Daniel Y. Fu, Stefano Ermon, Atri Rudra, and Christopher Ré. Flashattention: Fast and memory-efficient exact attention with IO-awareness. In Advances in Neural Information Processing Systems, 2022.arxiv.org/abs/2205.14135
- [7]Tri Dao et al. Flashattention-4: Algorithm and kernel pipelining co-design for asymmetric hardware scaling. In Proceedings of Machine Learning and Systems (MLSys), 2026. arXiv:2603.05451.arxiv.org/abs/2603.05451
- [9]DeepSeek-AI. FlashMLA: Efficient multi-head latent attention kernels. https://github.com/deepseek-ai/FlashMLA, 2025b.
- [10]DeepSeek-AI. Deepseek-v3.2: Pushing the frontier of open large language models, 2025c. arXiv:2512.02556.arxiv.org/abs/2512.02556
- [11]Weihua Du, Jingming Zhuo, Yixin Dong, Andre Wang He, Weiwei Sun, Zeyu Zheng, Manupa Karunaratne, Ivan Fox, Tim Dettmers, Tianqi Chen, Yiming Yang, and Sean Welleck. AdaExplore: Failure-driven adaptation and diversity-preserving search for efficient kernel generation, 2026. arXiv:2604.16625.arxiv.org/abs/2604.16625
- [12]HAN Lab Kernel Mafia. Developing GPU kernels with agentic loops. https://github.com/mit-han-lab/mlsys2026-flashinfer-contest, 2026. Technical report, with the per-operator phase prompts used for the competition runs.
- [13]Kimi Team. Kimi linear: An expressive, efficient attention architecture, 2025. arXiv:2510.26692.
- [14]METR. Measuring automated kernel engineering. https://metr.org/blog/2025-02-14-measuring-automated-kernel-engineering, 2025.
- [15]MIT HAN Lab. Kernel design agents: An agent-centric workflow for high-performance CUDA kernels. https://github.com/mit-han-lab/kernel-design-agents, 2026.
- [16]NVIDIA. AVO: Agentic variation operators for autonomous evolutionary search, 2026. arXiv:2603.24517.
- [18]Anne Ouyang, Simon Guo, Simran Arora, Alex L. Zhang, William Hu, Christopher Ré, and Azalia Mirhoseini. Kernelbench: Can LLMs write efficient GPU kernels?, 2025. arXiv:2502.10517.arxiv.org/abs/2502.10517
- [19]Sakana AI. The AI CUDA Engineer: Agentic CUDA kernel discovery, optimization and composition. https://sakana.ai/ai-cuda-engineer/, 2025.
- [20]Jay Shah, Ganesh Bikshandi, Ying Zhang, Vijay Thakkar, Pradeep Ramani, and Tri Dao. Flashattention-3: Fast and accurate attention with asynchrony and low-precision. In Advances in Neural Information Processing Systems, 2024.arxiv.org/abs/2407.08608
- [21]Songlin Yang, Jan Kautz, and Ali Hatamizadeh. Gated delta networks: Improving mamba2 with delta rule. In International Conference on Learning Representations, 2025.arxiv.org/abs/2412.06464
- [22]Zihao Ye, Yingyi Huang, Hongyi Jin, et al. CAKE: Compiler-agent co-design for frontier kernel evolution, 2026. arXiv:2608.12629.arxiv.org/abs/2608.12629
- [23]Yunxiang Zhang, Ping Yu, Jianyu Wang, Max Fan, Julian Reed, Azalia Mirhoseini, and Will Su. KernelBench-Verified: Do LLM-generated kernels actually beat PyTorch?, 2026. arXiv:2607.16241.arxiv.org/abs/2607.16241
- [24]Zijian Zhang, Rong Wang, Shiyang Li, Yuebo Luo, Mingyi Hong, and Caiwen Ding. Cud-aForge: An agent framework with hardware feedback for CUDA kernel optimization, 2025. arXiv:2511.01884.arxiv.org/abs/2511.01884