Notícias de IA · Amazon · chips
Amazon abre seus chips de IA Trainium ao código PyTorch
A Amazon diz que o TorchNeuron faz de seus chips um destino padrão do PyTorch, de modo que o código de IA possa sair do hardware da Nvidia sem ser reescrito.
A Amazon Web Services anunciou na segunda-feira, 19 de outubro de 2026, o TorchNeuron, um complemento gratuito que permite rodar nos chips Trainium da própria Amazon o código de IA escrito para PyTorch. O PyTorch é o kit de ferramentas que a maioria dos desenvolvedores usa para criar e rodar modelos de IA. A Amazon diz que cerca de 85% dos apps de IA são escritos com ele, informa o TechCrunch.
Para os rivais da Nvidia, o obstáculo sempre foi o código. O CUDA, a camada de software que controla os chips da Nvidia, é otimizado para o PyTorch há anos, e por isso o hardware da Nvidia é onde a maioria dos apps de IA acaba rodando. No exemplo básico do anúncio do PyTorch, a única mudança em um script de treinamento é enviar o modelo para um dispositivo chamado “neuron”; o modo compilado exige mais uma linha. A ferramenta cobre o Trainium2 e o Trainium3, tanto no modo passo a passo que os pesquisadores usam quanto no modo compilado usado em produção. O código vira público em beta, sob a licença Apache 2.0, em 20 de outubro, na PyTorch Conference.
Os chips da AMD e da Apple já oferecem uma mudança de configuração igualmente simples, observa o TechCrunch. A AWS vende o Trainium pelo preço em relação ao desempenho e por ter o resto da nuvem no mesmo lugar.
Para quem paga por clipe ou imagem de IA, mais chips rodando o mesmo código dão aos provedores dessas ferramentas mais margem para competir em custo.
Fontes
- Amazon: Amazon opens its AWS Trainium AI chips to every PyTorch developer, no rewrite needed. Verificado em
- PyTorch: Introducing TorchNeuron: native PyTorch backend for AWS Trainium. Verificado em
- TechCrunch: Amazon is taking its biggest shot at Nvidia yet with new TorchNeuron dev tool. Verificado em