Skip to content

Amazon goes after Nvidia's biggest advantage: its TorchNeuron tool moves AI apps onto Trainium chips by changing a single line

1 min read
Share
Amazon goes after Nvidia's biggest advantage: its TorchNeuron tool moves AI apps onto Trainium chips by changing a single line

Anyone following the artificial intelligence race even a little knows that Nvidia did not become the standard just because of its fast graphics chips. Its position is also held up by its CUDA software, which for years has made sure that apps written in PyTorch run on its hardware almost by themselves. That is exactly where Amazon Web Services is striking with its new TorchNeuron tool, unveiled on Monday.

The promise is simple: an AI app written in PyTorch can be moved onto Amazon's Trainium chip by changing one single line of configuration, without rewriting code. At least that is what the company claims. According to Amazon, about 85 percent of AI apps are written in PyTorch, an open-source tool created by Facebook and maintained by the Linux Foundation since 2022. This is not a small corner of the market, it is the majority of it.

Amazon is not the first to try. PyTorch apps can already be moved with a similar change onto AMD's graphics chips, and even onto Apple's chips. What Amazon is selling as the difference is the cloud around the chip. The company argues it is a better choice for startups than the so-called neoclouds, such as CoreWeave, Nebius and Firmus, many of which are tied to Nvidia and its hardware.

"You build on AWS. You don't bet on a single-purpose compute provider and then struggle to build everything else around it," said Colin Brace, vice president of technology software at AWS Annapurna Labs, the unit that makes Amazon's own chips. The price argument is more careful than it sounds: AWS does not claim the lowest price per token, but says the savings come because the same chip is used both for training and for running the finished model. As proof it cites Deepgram, an ElevenLabs competitor in voice AI, which after switching processed 70 percent more audio per dollar.

The bigger move is something else. Amazon says it will send changes into PyTorch itself, to clean out the code that ties the framework to specific chips. The company presents this as a service to the whole developer community, not just Trainium users. But when a chip vendor scrubs Nvidia dependencies out of the tool most people use, it is hard to believe in pure altruism. The whole point of PyTorch was to hide hardware headaches. We will see whether developers get more choice and cheaper options, or just a new configuration headache.

Brace did not beat around the bush: "The real story is removing the lock-in that stopped developers from having any real choice at all." If Amazon manages to deeply decouple PyTorch from Nvidia, the consequences for the giant will be far bigger than one more rival chip. For a smaller startup, from Skopje or from Belgrade, the question is practical: will it end up paying less, or just swap one landlord for another?