字节清华发布 CUDA-Agent:RL 生成高性能内核
AIFlow_ML · x · 2026-08-17
字节跳动与清华大学 SIA 联合发布了 CUDA-Agent,这是一个基于大规模智能体强化学习(RL)的系统,用于生成高性能 CUDA 内核。
主要特点:
- 性能表现:在 KernelBench 上取得 SOTA 结果,性能超越 Claude Opus-4.6 和 Gemini 3 Pro,在难度最高的案例中优势明显。
- 数据集:发布了 CUDA-Agent-Ops-6K 训练集(6000 个样本),构建流程包括从 PyTorch/Transformers 收集算子、LLM 融合任务生成以及规则过滤(确保可执行、确定性与非平凡性)。
- 开源内容:除了数据集,还开源了专家设计的 SKILL.md 和 Agent 环境,支持社区复现与研究。
「Infra」频道最新
- DGX Spark 运行 Qwen3.8-27B GGUF 命令示例 — ariG23498 · 2026-08-17
- llama-server 多模态推理遇报错:位置索引异常 — Gold-Drag9242 · 2026-08-17
- 手持 3090 显卡,如何入手本地部署 Qwen 3.8? — ozymandizz · 2026-08-17
- Qwen MLX 挑战赛上线,角逐本地模型最快推理 — corruptbytes · 2026-08-17
- CoreWeave 前三大客户占比降至 72%,微软依赖度大幅稀释 — AccBalanced · 2026-08-17
- MLX 框架仍活跃,社区已转向自研推理引擎 — AccBalanced · 2026-08-17