BL
MarkTechPost • 39일 전
바이트댄스·칭화대, 컴파일러 능가하는 CUDA 커널 생성 AI 'CUDA Agent' 공개
IMP 7/10
핵심 요약
바이트댄스 Seed와 칭화대 AIR 연구팀이 강화학습 기반 에이전트 시스템 'CUDA Agent'를 발표했습니다. 이 시스템은 대규모 언어모델을 학습시켜 컴파일러보다 빠른 GPU 커널을 작성하게 하는 것으로, 최신 모델들은 이미 정확한 CUDA 코드를 생성하지만 속도가 느리다는 문제를 해결하고자 합니다. KernelBench 벤치마크에서 기반 모델 Seed1.6은 74.0%의 통과율을 기록했습니다.
번역된 본문
바이트댄스 Seed와 칭화대 AIR이 'CUDA Agent'를 공개했습니다. 이는 에이전트형 강화학습(agentic reinforcement learning) 시스템으로, 대규모 언어모델(LLM)을 학습시켜 컴파일러를 능가하는 GPU 커널을 작성하게 만드는 기술입니다. 이 시스템이 겨냥하는 문제는 좁지만 고질적인 것입니다. 즉, 최신 프런티어 모델들은 이미 정확한(correct) CUDA 코드를 생성하지만, 단지 느린(slow) CUDA 코드를 생성한다는 점입니다. KernelBench 벤치마크에서 기반 모델인 Seed1.6은 74.0%의 통과율을 기록했습니다.
원문 보기 (영어)
ByteDance Seed and Tsinghua AIR have released CUDA Agent, an agentic reinforcement learning system that trains a large language model to write GPU kernels that beat a compiler. The gap it targets is narrow but stubborn: frontier models already produce correct CUDA, they just produce slow CUDA. On KernelBench, the base model Seed1.6 passes 74.0% […]
The post ByteDance Seed and Tsinghua AIR Introduces CUDA Agent: A Large-Scale Agentic RL System for CUDA Kernel Generation appeared first on MarkTechPost.