Field noteOct 30, 20266 min read1 source

OpenBMB open-sources ForgeKernel, a self-evolving GPU kernel optimization agent

In one minute

OpenBMB has open-sourced ForgeKernel, an Apache-2.0 workflow that uses a coding agent to iteratively optimize GPU kernels. It profiles a reference implementation, turns observations into gated implementation milestones, and promotes successful output into the next baseline. For teams with expensive custom operators, it makes optimization loops more inspectable and repeatable.

What changed: OpenBMB has open-sourced ForgeKernel, a self-evolving agent for GPU kernel optimization; the prior covers a different GPU kernel library.

What happened

OpenBMB published the ForgeKernel repository as an open-source project under the Apache-2.0 license. The project describes itself as a “self-evolving GPU kernel agent.” Its starting point is not a blank prompt to write CUDA-like code, but an existing reference kernel. ForgeKernel runs, times, and profiles that reference; uses the result to construct what it calls an architecture-first milestone curriculum; and has a coding agent implement the plan. A candidate only advances after it clears correctness, performance, and mechanism gates.

The central loop is deliberately recursive. An accepted kernel becomes the reference for the next run, so a later attempt must beat the previous best rather than merely improve on an original baseline. The README says this continues until the hardware offers no measurable further gain, defined as an improvement below the noise floor. It also describes a first milestone, M0, that settles costly architectural choices such as kernel decomposition, tile shapes, warp roles, and register and shared-memory budgets. Later milestones tune within that design. If measurement indicates that only a different structure can improve performance, the architecture agent can revise M0 instead of accumulating local workarounds.

ForgeKernel stores GPU-probe-certified observations in a hardware knowledge base organized by GPU architecture. The intended effect is that findings from one operator can inform later optimization work on the same hardware family. OpenBMB says the project is part of its Forge family alongside ForgeTrain and ForgeStencil. The repository includes generated kernels in its kernels/ directory, task definitions, scripts, tests, and an example that fuses a PyTorch RMSNorm operation from nine kernel launches into one CuTe DSL kernel.

The published setup is demanding and specific. The control machine needs Linux or macOS, Python 3.11 or later, Git, and the Claude Code CLI with a login or an ANTHROPIC_API_KEY; tmux is needed when a task lacks a reference. The GPU host can be local or reached through SSH, but must use an NVIDIA Hopper (sm90) or Blackwell (sm100) GPU and have a CUDA build of PyTorch. CuTe DSL tasks also need nvidia-cutlass-dsl; Nsight Compute is recommended, although the project can proceed using timing alone. The stated package version is forkernel 0.1.0.

For its example workflow, users clone the repository, create a Python virtual environment, install the package and recommended CuTe DSL dependency, configure a local or remote GPU in local.env, then create and run a task. A dry run prints the plan first. Runs can take hours, produce an artifact directory, and end with a report plus the best kernel. The README says the RMSNorm example stops when it reaches 1.2 times the reference or exhausts its budget; it reports a 5.2-times result against the cited PyTorch reference on a B200. Separately, OpenBMB reports that across 27 operators on Hopper and Blackwell, ForgeKernel matched or beat 18 of 22 expert-library kernels and all seven published kernels from three kernel-generation systems, with gains up to 170.6% and a 39.4% geometric mean. Those are the project’s own reported results.

What the archive adds

This is a new fact in a related infrastructure thread. Nvidia's open-source Run:ai release concerned orchestration of shared GPU workloads; ForgeKernel addresses performance inside an individual GPU operator. OpenBMB's MiniCPM-OCR release establishes recent activity from the same publisher, but covered a separate OCR and verification release.

Trend signal

ForgeKernel supports the pattern that goal-scoped loops replace manual continuation. Its loop has externally measured stop conditions: an explicit performance target, a finite budget, or an improvement that falls below measurement noise. The agent is therefore evaluated against a hardware result rather than its own judgment that the work is finished.

Why it matters

Custom kernels can be a meaningful cost and latency lever for teams running models at scale, but the work normally requires scarce GPU-performance expertise and an experimental discipline that is difficult to reproduce. ForgeKernel packages some of that discipline into an agent loop: establish a baseline, change one bounded design layer, test correctness, measure performance, retain evidence, and use the best artifact as the next baseline.

That does not mean teams should delegate production CUDA optimization blindly. The required hardware is narrow, a coding-agent credential is part of the setup, and the initial use case is for operators with enough repeated workload to justify hours of runs and subsequent review. Still, the repository offers a concrete model for turning an open-ended “make this faster” request into a sequence of checkable engineering objectives. For teams already paying for Hopper or Blackwell capacity, the released task structure and artifacts may be as useful as the benchmark claims.

What happens next

  • More teams will test ForgeKernel on their own high-volume operators rather than general benchmarks. (3 months, 65%). Watch for: public task configurations, reports, or kernels targeting application-specific workloads.
  • The project will broaden its hardware or task coverage only if its knowledge-base approach transfers beyond its stated Hopper and Blackwell environment. (6 months, 45%). Watch for: new architecture directories or documented support requirements.

Limits and uncertainty

The repository is the primary source for the release, requirements, workflow, and its reported measurements, but it is also the source of the comparative performance claims. The supplied material does not provide an independent replication, complete benchmark methodology, cost of coding-agent usage, or evidence that results transfer to a particular model-serving stack. “Matches or beats” is reported over a defined set of operators and libraries, not as a general claim that generated kernels will outperform expert work everywhere. The example’s B200 result is also relative to its stated PyTorch reference, whose suitability depends on the task being compared.

Skill to build now

Choose one repeated GPU-bound operation with a stable correctness test and record a reproducible baseline: input shapes, GPU type, runtime version, latency distribution, and acceptance threshold. Then separate architectural decisions from later tuning decisions and require every candidate to preserve correctness before comparing speed. That workflow is useful even without ForgeKernel, and it prepares a team to evaluate its artifacts rather than accepting an agent-generated kernel on benchmark output alone.

Reader takeaway

ForgeKernel is an open-source attempt to make GPU-kernel optimization a goal-bounded agent workflow rather than a one-off expert exercise. Its strongest practical contribution is the measured iteration loop and retained artifacts; its performance figures should be treated as vendor-reported results that teams can test against their own operators.

Open question

Can ForgeKernel’s hardware knowledge base and gated improvement loop deliver comparable gains on production-specific operators, rather than the published benchmark set?

Evidence / sources

  1. 1github.com/OpenBMB/ForgeKernel
All Field Notes

Next reads

Read next

Open the graph
  1. 01topicClaudeExplore the claude topic hub.
  2. 02topicClaude CodeExplore the claude-code topic hub.
  3. 03archiveField NotesOpen the latest editorial analysis.
  4. 04source ledgerSource LedgerInspect the public source evidence graph.
Who read this page?Machine requests, hidden until opened

Loading the privacy-safe route aggregate...

Open the JSON contract