Google and Meta launch TorchTPU, a native PyTorch backend for TPUs

PyTorch · x · 2026-10-22

PyTorch announced TorchTPU, an open-source native PyTorch backend for Google TPUs developed jointly by Google and Meta in partnership with PyTorch core maintainers, introduced by Bill Jia (Google Cloud) and Zongwei Zhou (Meta) at PyTorchCon 2026.

TorchTPU integrates with Hugging Face Transformers for modeling, TorchTitan for training, and vLLM and SGLang for inference, alongside agentic developer workflows demoed at the conference.

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

Original post →

More from Infra

Infra channel →