AWS Takes Aim at Nvidia’s PyTorch Advantage While Nvidia Expands Open-Weight Nemotron Models for Agents

AWS Takes Aim at Nvidia’s PyTorch Advantage While Nvidia Expands Open-Weight Nemotron Models for Agents

By James Dargan
Publication Date: 2026-10-20 16:10:00

Amazon Web Services has released TorchNeuron, a tool that lets AI startups move their applications onto Trainium, Amazon’s AI chip, in a direct challenge to the software advantage that has helped Nvidia become the default hardware for AI.

TorchNeuron works with PyTorch, the free, open source toolkit used to build, train and run AI models, and Amazon says it allows PyTorch apps to run on Trainium by changing a single line of configuration, with no code rewrite. Amazon estimates that about 85% of AI apps are written in PyTorch, where Nvidia’s CUDA software has long been deeply embedded. While AMD and Apple chips already offer similar porting options, AWS argues that Trainium also gives startups access to its entire cloud, positioning it as a stronger choice than Nvidia-aligned neoclouds such as CoreWeave, Nebius and Firmus.

Colin Brace, VP of Technology Software at AWS Annapurna Labs, said startups building on AWS have the infrastructure to move from prototype to production already in place. He added that Trainium’s value comes from its price-performance and from handling both training and inference on one platform, citing AI voice provider Deepgram, which he said now processes 70% more audio per dollar after moving to Trainium.

Amazon is also contributing code upstream to PyTorch to remove ties to specific chips, aiming to make the framework more hardware-agnostic for all developers. Brace said the bigger story is removing the lock-in that had denied…