CUDA-Agent
CUDA Agent: Large-Scale Agentic RL for High-Performance CUDA Kernel Generation
About
CUDA-Agent is the first known RL-trained model to surpass advanced models such as Claude Opus-4.6 and Gemini 3 Pro on high-performance CUDA kernel generation. It achieves state-of-the-art results on KernelBench, consistently outperforming the torch.compile baseline across difficulty levels, with especially strong gains on the hardest cases. To support the LLM-based CUDA generation community, we have released our training data, expert-designed SKILL.md and agent environment.
Open Source Health
- Stars
- 1,317
- Forks
- 136
- License
- Not stated
- Last commit
- 3 months ago
Resources & Links
Related Categories
Vendor
BytedTsinghua-SIA
The Joint SIA-Lab of Tsinghua AIR & Bytedance Seed - Scalable Intelligent Advanced Technology Joint Research Center
Quick Links
Open Source
More by BytedTsinghua-SIA
Related Products
The Algorithm
Source code for the X Recommendation Algorithm
Shared categories
Recommenders
TensorFlow Recommenders is a library for building recommender system models using TensorFlow.
Shared categories
RPG_KDD2025
This repository provides the code for implementing RPG described in our KDD'25 paper "Generating Long Semantic IDs in Parallel for Recommendation".
Shared categories
Pairec
A Go web framework for quickly building recommendation online services based on JSON configuration.
Shared categories
Textrank
TextRank implementation for Python 3.
Shared categories