L'ensemble de données AI-CUDA-Engineer-Archive est une collection de noyaux CUDA générés par l'IA, qui vise à faciliter la formation ultérieure de modèles open source et le développement de meilleurs modules de fonctions CUDA. L'ensemble de données contient plus de 30 000 noyaux CUDA, tous générés par des ingénieurs CUDA pilotés par l'IA, dont plus de 17 000 noyaux ont été vérifiés comme étant corrects, et environ 50% de noyaux surpassent l'environnement d'exécution natif de PyTorch.

Il comprend l'implémentation de référence PyTorch, les données d'analyse des performances PyTorch/NCU/Clang-tidy, plusieurs versions de noyau pour chaque tâche, des informations sur les erreurs et des données de comparaison des performances avec les environnements d'exécution natifs et compilés de PyTorch.