BytedTsinghua-SIA/CUDA-Agent
★ 1,160 stars
PythonCUDA Agent: Large-Scale Agentic RL for High-Performance CUDA Kernel Generation
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?