Sistema de IA da DoubleAI supera uma década de otimizações de engenheiros GPU
A DoubleAI desenvolveu WarpSpeed, um sistema de inteligência artificial capaz de otimizar automaticamente kernels GPU com desempenho superior ao de engenheiros humanos especializados. A empresa reoptimizou os kernels da biblioteca cuGraph (amplamente usada em aceleração por GPU) em três arquiteturas distintas — A100, L4 e A10G — e disponibilizou o código otimizado no GitHub.
Os resultados demonstram uma melhoria média de 3.6x relativamente aos kernels originais, com todos os algoritmos beneficiando de otimizações e mais de metade apresentando ganhos superiores a 2x. Diferentemente de aplicações anteriores onde a IA alcançou desempenho ao nível de especialistas (como em competições matemáticas), este feito é notável porque a engenharia de performance GPU apresenta desafios únicos: dados de treino escassos, validação complexa e interações profundas entre múltiplas escolhas de implementação.
A abordagem da DoubleAI combinou novas metodologias — um framework para aprender com volumes mínimos de dados, técnicas de verificação inovadora e estruturas de busca avançadas — para superar as limitações que fazem falhar sistemas de IA atuais neste domínio.