aifollow.news
Back Rohan Paul
Rohan Paul· @rohanpaul_ai · X· · Original publication time AI score54

UC Berkeley paper proposes reorganizing coding agent teams to optimize GPU kernelsMachine translation

AI introduction

Evolving Agent Teams uses a supervisor to diagnose stalls and rebuild agent teams, while a shared file system preserves lessons from earlier teams. On an MLA decode kernel, it reached 0.74 times FlashInfer's speed under the same token budget, versus 0.31 for a fixed four-role team. With more budget, it reached 1.21 times FlashInfer's speed and won on 36 of 47 production workloads.

Original post · Original

@rohanpaul_ai

New UC Berkeley paper shows that a supervisor that keeps reorganizing a team of coding agents can beat expert GPU kernels where single agents stall.

Coding agents make fast early progress on GPU kernels, then settle on easier designs and plateau far below expert code. On the MLA decode kernel, Claude Code stalled at 0.51 times FlashInfer's speed.

Evolving Agent Teams adds a supervisor that diagnoses each plateau and rebuilds the team, inventing roles like a profiler, a debugger, or a knowledge integrator. A shared file system keeps lessons from every past team.

Under the same token budget it reached 0.74 times FlashInfer on MLA decode, against 0.31 for a fixed 4-role team. With more budget it hit 1.21 times FlashInfer's speed, winning on 36 of 47 production workloads.

For long, hard optimization runs, have a supervisor restructure agent roles when progress stalls instead of running 1 fixed agent setup.

Source: Rohan Paul · x.com

Research
Found an error? Send a correction