A Amazon está ampliando seus investimentos em inteligência artificial ao apostar na formação da próxima geração de pesquisadores da área. Por meio do programa “Build on Trainium”, avaliado em US$ 110 milhões, a empresa oferece acesso ao chip Trainium, desenvolvido pela AWS especialmente para treinamento e operação de modelos avançados de IA.
A iniciativa já alcançou mais de 10 mil estudantes e pesquisadores de universidades como a University of California, Berkeley, o Massachusetts Institute of Technology, a Carnegie Mellon University e a University of Illinois Urbana-Champaign. O objetivo é acelerar pesquisas em áreas como aprendizado de máquina, saúde, computação quântica e ciência da computação.
Um dos principais desafios atuais da infraestrutura de IA é criar código capaz de explorar plenamente o desempenho dos chips especializados. Segundo a Amazon, muitos desenvolvedores ainda trabalham apenas com GPUs tradicionais, o que limita o potencial de novas arquiteturas especificamente projetadas para inteligência artificial.
Tarefas de IA
Na Universidade da Califórnia em Berkeley, duas equipes de pesquisa trabalham em soluções para otimizar os chamados “kernels”, rotinas computacionais responsáveis pela eficiência na execução de tarefas de IA. O grupo liderado pelo professor Christopher Fletcher desenvolveu o framework TeAAL, que automatiza a geração de código otimizado para o Trainium. Em testes, a tecnologia apresentou desempenho entre 1,4 e 1,6 vezes superior ao dos métodos tradicionais em tarefas de ajuste fino de modelos de IA.
Já os professores Sophia Shao e Alvin Cheung criaram o sistema Autocomp, que utiliza modelos de linguagem para gerar automaticamente código otimizado para o chip da Amazon. O próximo passo do projeto será expandir o sistema para abranger modelos completos, como Llama e Qwen.
Em outras universidades, o Trainium também vem sendo utilizado para acelerar descobertas científicas. No MIT, o pesquisador Brian Anthony emprega o chip no treinamento de modelos de análise de ultrassom 3D. Segundo a equipe, o desempenho foi 50% superior ao de GPUs tradicionais, reduzindo o tempo de treinamento de meses para semanas.
Circuitos quânticos
Na University of California, Los Angeles, o professor Jens Palsberg utiliza a plataforma para simular circuitos quânticos, uma das áreas mais desafiadoras da computação moderna.
Outro destaque do programa é o caráter open source. Todas as pesquisas desenvolvidas serão disponibilizadas publicamente, permitindo que as melhorias feitas no Trainium beneficiem toda a comunidade de desenvolvedores e pesquisadores de inteligência artificial.
Com a iniciativa, a Amazon busca não apenas fortalecer sua divisão de chips para IA, mas também ampliar o acesso a tecnologias avançadas capazes de impulsionar avanços em áreas como saúde, robótica e agentes inteligentes.
