Датасет · BytedTsinghua-SIA
CUDA-задачи для агентного RL
BytedTsinghua-SIA/CUDA-Agent-Ops-6K
6 тыс. синтезированных задач уровня операторов для генерации и оптимизации CUDA-ядер.
Что внутри
Это подборка из 6 тыс. синтезированных тренировочных задач operator-level для CUDA. Они предназначены для масштабного агентного RL и должны быть исполняемыми. Данные текстовые, на английском; выпущены в рамках проекта CUDA-Agent.
Что можно сделать
- Обучить LLM-агента генерировать CUDA-ядра
- Провести RL-дообучение оптимизации ядер
- Оценить исполняемость и качество кода агента
- Задачи
- Генерация текста
- Языки
- Английский
- Размер
- 1K–10K записей
- Скачиваний за месяц
- 298
- Лайков
- 94
Как загрузить
from datasets import load_dataset
ds = load_dataset("BytedTsinghua-SIA/CUDA-Agent-Ops-6K")