Amazon and Google Launch Tools to Challenge Nvidia's AI Chip Dominance with PyTorch Support

13 Sources

Share

Amazon and Google released TorchNeuron and TorchTPU, open-source tools that let AI developers run PyTorch code on their custom AI chips with minimal changes. The move directly challenges Nvidia's market dominance by making it easier to switch from Nvidia's GPUs to alternative hardware without rewriting applications.

Amazon and Google Target Nvidia's Software Lock-In

Amazon Web Services and Google have launched competing open-source tools that allow AI developers to run PyTorch code on their custom AI chips without rewriting applications

1

2

. Amazon announced TorchNeuron on Monday, followed by Google's TorchTPU release on Tuesday, marking a coordinated effort to challenge Nvidia's dominance in AI hardware

1

. PyTorch, used by approximately 85% of AI apps, has been deeply integrated with Nvidia's CUDA software infrastructure over more than a decade, making Nvidia GPUs the default choice for AI workloads

2

4

.

How TorchNeuron and TorchTPU Work

Both tools dramatically simplify the process of porting AI applications to alternative hardware. TorchNeuron allows developers to run PyTorch code on Amazon's Trainium chips by changing just one line of configuration, with no code rewrite required

2

4

.

Source: GeekWire

Source: GeekWire

Google's TorchTPU enables similar functionality for Google's Tensor Processing Units, requiring only a simple configuration change to move existing PyTorch apps to TPUs and Google Cloud

1

. TorchNeuron entered public beta on 20 October under the Apache 2.0 licence and supports both Trainium2 and Trainium3 chips

5

. Colin Brace, VP of Technology Software at AWS Annapurna Labs, stated the move aims to reduce reliance on Nvidia by "meeting customers where they're at"

4

.

Strategic Partnerships Strengthen Google's Position

Google worked directly with Meta, Hugging Face, Inferact, and RadixArk to ensure TorchTPU integrates seamlessly with popular AI development tools

1

.

Source: SiliconANGLE

Source: SiliconANGLE

Meta, PyTorch's original creator, now runs SGLang-TorchTPU in production, serving AI models on Google's TPUs

3

. Jana van Greunen, director of PyTorch engineering at Meta, explained that Meta "cannot be constrained by a single hardware architecture" and that TorchTPU "enables us to deploy our PyTorch workloads to Google's Tensor Processing Units"

3

. Google also launched a dedicated TPU agent suite to help AI developers migrate applications and fix errors, plus updated its LiteRT tool for running AI on phones, laptops and embedded devices to work better with PyTorch apps

1

.

Performance Claims and Early Adoption

Inferact reported that a tuned vLLM-TorchTPU setup serving Moonshot AI's 2.8-trillion-parameter Kimi K3 model achieved over 700 tokens per second for a single user on seventh-generation Ironwood TPUs, compared to 452 tokens per second on 16 Nvidia GB200s

3

. Amazon claims more than 30 early customers spanning robotics, world models, and voice AI are using TorchNeuron, with the switch to Amazon's Trainium chips resulting in steep cost savings

4

. Deepgram, an AI voice app provider, reportedly processes 70% more audio per dollar at production-grade accuracy after moving to Trainium using native PyTorch support

2

. Early users also include Cartesia, the Lieber Institute for Brain Development, and Uber

5

.

Amazon Pushes for Hardware-Agnostic PyTorch

Amazon is going beyond simply creating an open-source tool for its own chips. The company announced it will contribute code to the main PyTorch project to decouple various features from Nvidia's chips, making PyTorch more hardware-agnostic by default

1

2

. Brace stated that "the real story is removing the lock-in that kept developers from having a real choice at all"

2

. This approach extends Amazon's strategy beyond promoting its own AI infrastructure to fundamentally reshaping how PyTorch interacts with AI hardware across the industry.

Market Implications and Future Outlook

The coordinated launches represent the most significant challenge yet to Nvidia's market dominance in AI computing. Amazon's custom chip business has reportedly passed a $25B annual revenue run rate, with CEO Andy Jassy projecting it could reach $50B

5

. However, both companies acknowledge that making PyTorch code portable doesn't guarantee developers will automatically abandon Nvidia's AI hardware

1

. The first release of TorchNeuron has limitations, with approximately 22,000 PyTorch tests skipped and some features like dynamic input shapes not yet supported

5

. RadixArk plans to add support for additional models including DeepSeek V4.1 Flash and Reflection Beam, working toward Day 0 TPU support for new open models by the first quarter of 2027

3

.

Source: Tom's Hardware

Source: Tom's Hardware

Watch for how quickly AI developers adopt these tools and whether the promised cost savings materialize at scale, as this will determine if Amazon and Google can truly reduce Nvidia's software advantage in the AI chip market.

Today's Top Stories

© 2026 TheOutpost.AI All rights reserved