跳到正文
热点事件持续更新

Ningyuan Xi等发布CUDA内核生成数据集KernelMosaic

1 篇报道1 个报道来源53 分钟前更新

先了解这件事

AI 综述

Ningyuan Xi 等研究者发布开源数据集与智能体框架 KernelMosaic,将多样化 CUDA 任务与经验证的解决方案和轨迹配对,用于训练大语言模型生成 CUDA 内核。 论文称,用该数据集对小型开源模型做监督微调后,模型在 KernelBench L1/L2/L3 上的正确率分别从 4%、0%、0% 提升至 88%、82%、16%,且各层级快于 PyTorch eager 的内核比例均上升。训练实验还显示,基于精选轨迹的监督显著优于直接响应。

AI 根据报道生成 · 18 分钟前更新

报道时间线

沿着报道,了解事件的不同侧面。

11月14日
  1. arXiv cs.AI
    KernelMosaic:面向 CUDA 内核生成的开源数据集与智能体框架

    KernelMosaic 是一个开源数据集与智能体框架,将多样化 CUDA 任务与经验证的解决方案和轨迹配对,用于训练大语言模型生成 CUDA 内核。在小型开源模型上做监督微调后,KernelBench L1/L2/L3 的正确率分别从 4%、0%、0% 提升至 88%、82%、16%,且各层级快于 PyTorch eager 的内核比例均上升。训练实验表明,基于精选轨迹的监督显著优于直接响应。

本事件热度走势

还没有足够的连续观测数据,暂不绘制趋势。