Google open-sources TorchTPU with Meta, letting PyTorch workloads run natively on TPUs

mark_k · x · 2026-10-21

Google, built with Meta's PyTorch team, has open-sourced TorchTPU, letting existing PyTorch workloads run natively on Google TPUs with minimal code changes across training and inference. It integrates with Hugging Face, vLLM and SGLang; RadixArk says its SGLang integration already runs at large-scale production in Meta, with the same serving code working across GPUs and TPUs including prefix caching and speculative decoding. Meta's PyTorch engineering director: "We cannot be constrained by a single hardware architecture."

Related event: Google and Meta Open-Source TorchTPU, Bringing Native PyTorch Support to TPU(15 posts)→

Original post →

More from Infra

Infra channel →