El sistema de IA de DoubleAI supera el rendimiento de ingenieros de GPU expertos

DoubleAI presentó WarpSpeed, un sistema de inteligencia artificial capaz de optimizar automáticamente kernels de GPU con un desempeño superior al de ingenieros humanos especializados. El sistema reoptimizó los kernels de cuGraph, una biblioteca ampliamente utilizada para aceleración por GPU, en tres arquitecturas diferentes: A100, L4 y A10G. La versión optimizada se encuentra disponible en GitHub para su implementación inmediata sin cambios en el código existente. Los resultados muestran una mejora promedio de 3.6x respecto a los kernels originales, con beneficios en la totalidad de los algoritmos y mejoras superiores a 2x en el 55 por ciento de los casos. Este logro es particularmente significativo porque la ingeniería de rendimiento en GPU enfrenta desafíos únicos: escasez de datos de entrenamiento, validación compleja y múltiples interdependencias entre decisiones de implementación. Sistemas de IA actuales, incluyendo modelos destacados de generación de código, fallan consistentemente al intentar producir implementaciones correctas en este dominio especializado. DoubleAI abordó estos desafíos mediante innovaciones técnicas novedosas: un framework que aprende de datos extremadamente limitados, metodologías de verificación sin verdad base, y estrategias avanzadas de búsqueda para navegar cadenas largas de optimizaciones interconectadas.