Skip to main content
Google & Amazon logos flank Nvidia's logo, symbolizing their fight for AI app developers using PyTorch.

Editorial illustration for Google Joins Amazon in Nvidia Fight Over AI App Developers and PyTorch

Google Challenges Nvidia With PyTorch for TPUs

Google Joins Amazon in Nvidia Fight Over AI App Developers and PyTorch

• 4 min read

Google released TorchTPU as an open-source project on Tuesday, giving developers a version of PyTorch built to run on the company's own Tensor Processing Units instead of Nvidia's GPUs. The move puts Google alongside Amazon, which made a nearly identical announcement a day earlier with TorchNeuron, its PyTorch variant for Amazon's Trainium chips. Both releases target the same soft spot in Nvidia's business: PyTorch, the open-source framework that something like 85% of AI applications are built on.

PyTorch came out of Meta, back when it was still Facebook, in 2016. At that point Nvidia's GPUs were already the default hardware for AI research, so a lot of PyTorch's underlying architecture got built around Nvidia's chips specifically. Meta handed the project to the Linux Foundation in 2022, but the Nvidia-shaped bones of the framework stuck around.

Google's pitch with TorchTPU is simplicity: existing PyTorch apps can shift to TPUs, and to Google Cloud, through a configuration change rather than a rewrite. Amazon is making the same case for Trainium and AWS. But Amazon isn't stopping at its own framework fork.

On Tuesday, Google announced that TorchTPU, its long-awaited version of PyTorch, is now available as an open-source project. With this, Google joins Amazon in trying to break Nvidia’s stranglehold on the AI industry, where most AI apps are written by default to run on Nvidia’s chips, or GPUs.

Why this matters

TorchTPU doesn't change what you write, it changes what you can run it on without rewriting anything. That's the real target here: the default assumption that PyTorch code means Nvidia GPU. Google and Amazon both know they can't out-CUDA Nvidia on tooling maturity, so they're going after the layer underneath it, the 85% of AI apps that already speak PyTorch. If TorchTPU works as advertised, a startup training on Google's TPUs stops being a niche choice and becomes a default alternative, not a migration project.

For founders burning cash on GPU allocation waitlists, that's the headline, not the open-source announcement itself. For researchers, the question is whether TorchTPU actually matches CUDA's performance quirks or just approximates them, because PyTorch's architecture was built around Nvidia hardware since 2016 and that history doesn't disappear with a new backend. We'd watch benchmarks closely before assuming parity.

Nvidia's moat was never just chips, it was the years of software optimization baked into that relationship. This is the first real attempt to route around it at scale.

Common Questions Answered

What is TorchTPU and how does it differ from standard PyTorch?

TorchTPU is Google's open-source version of PyTorch that is specifically optimized to run on Google's Tensor Processing Units (TPUs) instead of Nvidia's GPUs. The key difference is that TorchTPU allows developers to run existing PyTorch code on TPUs without rewriting their applications, effectively removing the default assumption that PyTorch means Nvidia hardware.

Why did Google and Amazon release PyTorch variants on the same week?

Both Google and Amazon released PyTorch variants—TorchTPU and TorchNeuron respectively—to challenge Nvidia's dominance in the AI chip market. These releases target the same vulnerability in Nvidia's business model: approximately 85% of AI applications are built on PyTorch, creating a default preference for Nvidia's GPUs that both companies aim to disrupt.

How does TorchTPU address Nvidia's stronghold in the AI industry?

TorchTPU targets the layer underneath CUDA tooling by allowing developers to run the 85% of existing PyTorch applications on Google's TPUs without modification. By making TPUs a viable default alternative rather than a niche choice, Google undermines Nvidia's competitive advantage that stems from PyTorch's de facto association with Nvidia hardware.

What is the strategic advantage of Google and Amazon not trying to compete with Nvidia's CUDA?

Rather than attempting to match Nvidia's mature CUDA tooling ecosystem, Google and Amazon chose to focus on the PyTorch layer above it, where they can leverage existing code compatibility. This strategy allows them to capture AI developers and applications without requiring them to rewrite code or learn new frameworks, making their chips more immediately accessible.

LIVE22:07Elevenlabs releases two speech-to-speech models for real-time AI conversations