Der AI-CUDA-Engineer-Archive-Datensatz ist eine Sammlung von durch KI generierten CUDA-Kerneln, die das anschließende Training von Open-Source-Modellen und die Entwicklung besserer CUDA-Funktionsmodule erleichtern soll. Der Datensatz enthält mehr als 30.000 CUDA-Kernel, die alle von KI-gesteuerten CUDA-Ingenieuren generiert wurden. Bei mehr als 17.000 Kerneln wurde die Richtigkeit verifiziert, und etwa 50% der Kernel übertreffen die native Laufzeit von PyTorch.

Es umfasst die PyTorch-Referenzimplementierung, PyTorch/NCU/Clang-tidy-Leistungsanalysedaten, mehrere Kernelversionen für jede Aufgabe, Fehlerinformationen und Leistungsvergleichsdaten mit nativen und kompilierten PyTorch-Laufzeiten.