Attention backward.
256-wide heads, sliding windows, and packed documents.
vs production baseline

GPU KERNELS · REINFORCEMENT LEARNING
RL environments for agents that write GPU kernels. Evaluate against expert production kernels, with sealed graders, expert-reviewed solutions, and scored trajectories.



01 / The environments
Each environment ships with a sealed grader, an expert-reviewed solution and scored agent trajectories.
256-wide heads, sliding windows, and packed documents.
Long-context attention over a growing KV cache.
ZIP · 82.4 MB. B300 required to run.
Opus 5.5 at max effort, up to 120 minutes per attempt. Attention backward counts 15 graded attempts across three task versions. Chunked prefill was screened on earlier task versions. Expert-reviewed solutions passed 3/3 reruns on B300.
02 / Distributed training
Each environment asks an agent to make a real mixture-of-experts model train faster on 8× A100 without changing the training math. Every step is checked against a reference run, and the bar is the best published training recipe.
128 experts with top-8 routing, expert parallel layout and grouped kernels.
256 experts, compressed sparse attention and hyper-connections.
ZIP · 9.8 MB. 8× A100 80 GB required to run.
Opus 5.5 at max effort, up to 120 minutes per attempt. Passes count the first 5 graded attempts on each task version, and every attempt was correct. Verified solutions passed 3/3 sealed grades on 8× A100. Both models run at reduced depth so full training fits 8× 80 GB GPUs.
03 / The evidence
Real attempts, real gradings and real cheats, all on NVIDIA B300. Rewards follow speed against the expert kernel, so agents get a learning signal on every attempt.
FIG 1 / ATTENTION BACKWARD
FIG 2 / REWARD SIGNAL
FIG 3 / GRADER INTEGRITY
04 / How it's graded
The agent writes a kernel. Everything that decides its reward runs where the agent cannot reach it.
The grader, hidden sizes and expert code never enter the agent's workspace.
Each grade adds input sizes the agent never saw, next to the public ones.
New random values on every timed call, so cached or stale outputs fail.
The expert kernel is timed in the same run on the same B300.
Correct: log2 of speed against the expert. Wrong or cheating: −24.
05 / The benchmark
KernelBench asks models to beat PyTorch. Our environments ask them to beat expert production kernels that serve real models.
Best fast_1 vs PyTorch: o1 in the 2025 paper, STARK agents by March 2026.
Paper ↗ · Leaderboard ↗Max effort, up to 120 min per attempt, B300.
See the environment →Different models, metrics and baselines. Shown for context, not as a ranking.
KernelBench details from the official repository and paper.
For AI labs