SAVRN
Search Contact SAVRN

Organization · Verified on Hugging Face

Sakana AI

SakanaAI

We are a Tokyo-based R&D company on a quest to create a new kind of foundational AI model based on nature-inspired intelligence.

Models in Library0
Datasets in Library1
Models on Hugging Face49
Followers940

Datasets

We release The AI CUDA Engineer archive, a dataset consisting of approximately 30,000 CUDA kernels generated by The AI CUDA Engineer. It is released under the CC-By-4.0 license and can be accessed via HuggingFace and interactively visualized here. The dataset is based on the Kernel tasks provided in KernelBench and includes a torch reference implementation, torch, NCU and Clang-tidy profiling data, multiple kernels per task, error messages and speedup scores against torch native and compile runtimes. We envision that this dataset can enable post-training of open-source models to perform better CUDA-enabling modules. This includes offline Reinforcement Learning, preference optimization, and…

Publicly accessible cc-by-4.0