Датасет · SakanaAI
Архив CUDA-ядер от AI CUDA Engineer
SakanaAI/AI-CUDA-Engineer-Archive
Около 30 000 CUDA-ядер, сгенерированных агентом, с эталонными реализациями и метриками.
Что внутри
Архив содержит примерно 30 000 CUDA-ядер, сгенерированных AI CUDA Engineer. Он основан на задачах KernelBench и включает эталонную реализацию на torch, а также данные torch, NCU и Clang-tidy. Датасет выпущен под лицензией CC-BY-4.0, доступен на Hugging Face и в интерактивной визуализации. Модальности: таблицы и текст.
Что можно сделать
- Обучить модель генерации CUDA-ядер по описанию задачи.
- Дообучить LoRA или агента для оптимизации CUDA-кода.
- Сравнить агентов на задачах в духе KernelBench.
- Размер
- 10K–100K записей
- Скачиваний за месяц
- 11,8 тыс.
- Лайков
- 239
Модели, обученные на нём
Как загрузить
from datasets import load_dataset
ds = load_dataset("SakanaAI/AI-CUDA-Engineer-Archive")