AWS lança TorchNeuron para levar apps PyTorch aos chips Trainium

A Amazon Web Services lançou na segunda-feira, 19 de outubro de 2026, o TorchNeuron, uma ferramenta para desenvolvedores criada para ajudar startups de IA a migrar seus aplicativos para o Trainium, o chip de IA desenvolvido internamente pela Amazon. O lançamento mira diretamente a barreira de software que fez da Nvidia o hardware padrão para a maioria das cargas de trabalho de IA.

O TorchNeuron foi construído em torno do PyTorch, o framework gratuito e de código aberto que desenvolvedores usam para criar, treinar e executar modelos de IA. De acordo com a Amazon, um aplicativo escrito em PyTorch pode rodar no Trainium após alterar apenas uma linha de configuração, sem reescrever nenhum código. Como a ferramenta foi adaptada do próprio PyTorch, desenvolvedores que já conhecem o framework não devem enfrentar uma curva de aprendizado.

Enfrentando o domínio do CUDA

A Amazon estima que cerca de 85% dos apps de IA são escritos em PyTorch. Originalmente criado pelo Facebook, o framework está sob a guarda da Linux Foundation desde 2022. A Nvidia passou anos garantindo que sua pilha de software CUDA funcionasse perfeitamente com o PyTorch, tornando na prática suas GPUs o alvo padrão do framework. A Amazon não é a primeira a tentar afrouxar esse vínculo: usuários de PyTorch já podem trocar para GPUs da AMD ou para chips da Apple com uma simples mudança de configuração.

A AWS argumenta que sua proposta vai além da facilidade de portabilidade, porque migrar para o Trainium também dá acesso ao restante da sua nuvem. A empresa apresenta a oferta como uma opção mais forte para startups do que as chamadas neoclouds, como CoreWeave, Nebius e Firmus, muitas das quais dependem de hardware da Nvidia. Colin Brace, vice-presidente de Software de Tecnologia da AWS Annapurna Labs, unidade responsável pelos chips customizados da Amazon, disse que usar um único chip para treinamento e inferência melhora a economia geral, mesmo que a AWS não ofereça o menor preço por token.

Brace citou a empresa de IA de voz Deepgram como exemplo. Segundo ele, a Deepgram migrou para o Trainium usando o suporte nativo ao PyTorch e hoje processa 70% mais áudio por dólar, com precisão de nível de produção.

Levando o PyTorch rumo à neutralidade de hardware

A Amazon também afirmou que contribuirá com alterações para o projeto principal do PyTorch, trabalhando com a comunidade para remover o código que vincula o framework a chips específicos. O objetivo declarado é um PyTorch mais agnóstico em relação ao hardware para todos os desenvolvedores, e não apenas para clientes do Trainium.

Ainda está para ser visto se isso trará mais escolhas e custos menores ou apenas acrescentará complexidade de configuração que o PyTorch deveria ocultar. Se a Amazon conseguir desvincular o framework da Nvidia, as consequências para a líder em GPUs podem ir muito além da concorrência comum em chips. "A história real é remover o aprisionamento que impedia os desenvolvedores de ter uma escolha de verdade", disse Brace.