Skip to main content
TTT-Discover uses RL at inference to optimize GPU kernel speed, outperforming human experts by 2x [threads.com](https://www.t

Editorial illustration for TTT-Discover uses inference-time RL to double GPU kernel speed vs experts

AI Learns to Optimize GPU Kernels During Inference

TTT-Discover uses inference-time RL to double GPU kernel speed vs experts

Updated: 2 min read

Forget crafting AI to remember. Build it to forget. That's the stark proposition from researchers behind TTT-Discover.

Their system churns out GPU kernels twice as fast as the best human versions. Its neural network is a temporary scaffold, a throwaway tool. Its sole purpose is to discover one perfect piece of code.

Then you toss the model. This isn't about creating intelligence. It's about using a neural network as a consumable fuel for a single, high-stakes task.

“TTT-Discover uses the same training stack (GPUs, rollout workers, optimizers, checkpointing).”

The implications cut against the grain. Modern AI chases bigger, general models that learn and retain. TTT-Discover proposes the opposite: a specialist that forgets.

All value is locked in that one perfect artifact it finds before deletion. That’s a different efficiency. It trades broad, average competence for a single, unbeatable solution.

For brutally expensive problems—where every nanosecond on a GPU cluster counts—that’s a radical trade. The future of optimization might not be a smarter AI. It might just be a disposable one.

Common Questions Answered

How does TTT-Discover differ from traditional reinforcement learning approaches?

Unlike standard reinforcement learning that aims to create a generalist policy performing well on average across tasks, TTT-Discover focuses on finding the absolute best solution to a specific problem. The method treats the model as a means to discover an optimal artifact, such as an ultra-efficient GPU kernel, by continuously learning and updating during the test phase itself.

What specific performance improvements did TTT-Discover achieve in GPU kernel engineering?

TTT-Discover demonstrated significant speed improvements in GPU kernel performance, reducing execution times by nearly 50% compared to human expert implementations. For instance, on the H100 GPU, the method generated a kernel running at 1161 μs, which was substantially faster than the previous best human-created kernel at 1371 μs.

What makes the test-time training approach of TTT-Discover unique?

TTT-Discover introduces a novel approach called 'test-time training' where the model continues to learn and refine its solution during the inference phase for a specific problem. By using an entropic objective that prioritizes finding the single best solution rather than average performance, the method allows the model to dynamically update its weights and internalize the specific structure of the task at hand.

LIVE03:21OpenAI's Miles Wang in Talks for USD 2B AI Drug Discovery Startup