Refacto AI

Industry story

Hawkeye Enables AI Agents to Write Hardware-Optimized GPU Kernels

agents coding-agents evals gpu-supply inference

Researchers from Harvard, Stanford, Together AI, and Caltech released Hawkeye, an open-source framework that helps AI coding agents write highly optimized GPU kernels — low-level software routines that maximize hardware performance — with minimal human expert input. The key innovation is a curated taxonomy of unit tests pairing human-written example kernels with hardware-specific profiling metrics, giving agents concrete guidance on architecture-specific optimizations. Tested across NVIDIA Ampere, Hopper, Blackwell, and AMD MI350 GPUs, Hawkeye matched or exceeded expert-tuned vendor libraries on established workloads and achieved an 18.9× speedup over expert-authored code on emerging attention variants. The paper argues this illustrates a broader pattern: small amounts of well-structured human-curated knowledge can allow AI systems to exceed the best human-written baselines.

Analysis

Showing the shorter version.

Hawkeye is an open-source framework out of Harvard, Stanford, Together AI, and Caltech that lets AI coding agents write hardware-optimized GPU kernels. A kernel is the low-level routine that determines how fast a chip actually executes matrix math. Writing a good one has always required a scarce, expensive human. Hawkeye claims to match or beat vendor-tuned libraries across NVIDIA Ampere, Hopper, and Blackwell, plus AMD's MI350, and posts an 18.9x speedup over expert-written code on emerging attention variants.

The 18.9x number is real but narrow. It lands on exotic attention variants, linear attention and state-space model scans, where torch.compile can't fuse the pattern and hand-optimization was the only option. Most production inference volume still runs plain softmax attention, where FlashAttention-3, cuBLAS, and cuDNN are already close to the hardware floor. The benchmark suite was also curated by the same team that built the framework. Not fraud, but selection pressure. The win looks spectacular exactly where hand-tuning was hardest, and unremarkable where the volume actually lives.

The AMD angle is what makes this strategically interesting anyway. AMD's MI350 has real FLOPs and a chronic software gap. That gap is the primary reason NVIDIA commands the pricing it does. A framework that writes competitive AMD kernels attacks that gap at precisely the moment AMD is trying to take inference share. If the MI350 results hold outside the benchmark suite, the per-token cost advantage that well-staffed inference teams hold over everyone else starts compressing, and that flows straight into pricing at Together AI, Fireworks, and anyone currently underselling OpenAI and Google.

The less obvious contribution is the taxonomy of unit tests. The framework pairs human-written example kernels with hardware-specific profiling metrics, compressing expert intuition into a machine-readable scaffold that a model can grade its own work against. That is a portable idea, and it is the mechanism behind the speedup.

Before leaning on this, run one eval: take a non-standard attention kernel you already ship, generate the Hawkeye version, and diff outputs across your full input distribution including ragged batch edges. The speedup is easy to measure. Correctness is what ends careers, and kernel bugs are silent until live traffic finds them.

The call: By the March 2027 MLPerf Inference round, at least one inference provider or open-source serving stack (Together AI, Fireworks, vLLM, or SGLang) will ship or publicly demo Hawkeye-generated kernels for a non-standard attention variant in a production or near-production path. Together AI co-authored the framework and runs a commercial inference business where margin is per-token compute cost. It has both the means and the motive. The failure mode is silent correctness regressions forcing a quiet rollback, not lack of interest.

Also covered this issue

Comments