TorchTPU

Google open-sources TorchTPU, a native PyTorch stack for Cloud TPUs co-developed with Meta

Google Cloud Blog published this on Oct 20, 2026 (source: cloud.google.com).

Google announced at PyTorch Conference 2026 that TorchTPU, its native PyTorch stack for TPUs co-developed with Meta's PyTorch team, is now open source on GitHub. It lets existing PyTorch workloads run on TPUs with minimal code changes, using eager mode for debugging and torch.compile for production performance. Google says models can be trained and served on TPUs with tools including vLLM, SGLang, llm-d, TorchTitan, Ray, TorchRec and Hugging Face, where Transformers v5 deploys to TPUs with device_map="tpu".