← The log

BytedTsinghua-SIA/CUDA-Agent

★ 1,160 stars

Python

CUDA Agent: Large-Scale Agentic RL for High-Performance CUDA Kernel Generation

View on GitHub

27 Aug 2026 · 0:30
This AI Writes CUDA Kernels Faster Than Claude 🤯

Transcript

Writing optimized CUDA kernels manually usually means weeks wrestling with GPU memory, thread sync, and performance bottlenecks—only to end up with code slower than hardware limits. CUDA-Agent automates kernel generation using reinforcement learning. The AI agent explores the optimization search space and generates high-performance kernels that rival expert-written code. 1,160 stars, backed by ByteDance and Tsinghua University. Would you trust AI to write your GPU code?