Skip to content
aiqbench takeChips & Infra

aiqbench analysis

Google and Meta's open-source TorchTPU backend could lower the cost of moving PyTorch models off Nvidia GPUs, but the claims are still mostly vendor and keynote statements. Readers should watch for independent benchmarks, TPU availability for smaller teams, and how much of the "no code changes" promise holds for real production models. The usage growth figures cited come from a single platform, so treat them as directional.
Source coveringInfoQ

PyTorch Conference 2026: Google and Meta Open-Source TorchTPU for Native PyTorch on TPUs

InfoQ5 min read

Google and Meta open-sourced TorchTPU, a native PyTorch backend for Google's TPUs, at PyTorch Conference North America 2026, and Meta said it plans to move larger LLM workloads onto TPUs with it. AWS released TorchNeuron for Trainium the same week, while developers questioned TPU availability and where vendor lock-in now sits. By Andrew Hoblitzell

At PyTorch Conference North America 2026 in San Jose, Google and Meta open-sourced TorchTPU, a native PyTorch backend for Google's Tensor Processing Units (TPUs). Meta, which created PyTorch in 2016, plans to move larger-scale LLM workloads onto TPUs with it, Meta software engineer Zongwei Zhou said in the October 21 keynotes. TorchTPU replaces PyTorch/XLA, a lazy-tensor bridge from 2020 that, according to the announcement, "required intrusive source-code modifications." The new backend aims to make porting a model as simple as changing model.to("cuda") to model.to("tpu"). "With TorchTPU, TPUs join GPUs and CPUs as a native option for PyTorch," Google Cloud wrote in its announcement, co-authored…

Excerpt shown under the publisher's RSS terms · 104 words · full text and figures are on the publisher's site