字节 Seed 与清华 AIR 发布 CUDA Agent,KernelBench 上内核快于 torch.compile 的比例达 96.8%
ByteDance Seed and Tsinghua AIR Introduces CUDA Agent: A Large-Scale Agentic RL System for CUDA Kernel Generation
阅读原文
本站未展示全文,请前往来源网站阅读。
AI 导读
字节跳动 Seed 与清华 AIR 发布 CUDA Agent 智能体强化学习系统,把大模型放进带性能剖析、正确性校验和沙箱权限锁定的 CUDA 开发环境,用 PPO 训练 150 步,在 KernelBench 250 项任务上取得 98.8% 通过率和 96.8% 快于 torch.compile 的比例,几何平均加速 2.11×。
来源:MarkTechPost · marktechpost.com