CUDA Agent: Large-Scale Agentic RL for High-Performance CUDA Kernel Generation
IntermediateWeinan Dai, Hanlin Wu et al.Feb 27arXiv
CUDA Agent is a training system that teaches an AI to write super-fast GPU code (CUDA kernels) by practicing, testing, and getting rewards for correct and speedy results.
#CUDA kernel generation#agentic reinforcement learning#PPO actor-critic