Google open-sources TorchTPU for native PyTorch on TPUs

Google said TorchTPU, its native PyTorch stack for TPUs, is now open source. The company said existing PyTorch workloads can run on TPUs with small code changes, including through vLLM, SGLang, Hugging Face, and TorchTitan, and that it built the stack with Meta's PyTorch team. The PyTorch project said TorchTPU is in active development and is already used in production for some Google Cloud customers.

/ Sources

/ About this story

Compiled by Venture Atlas from the sources above, using automated AI-assisted research. This is a summary of reporting published elsewhere, not original reporting - follow the source links for the full account. See our editorial standards.

Something wrong here? Email apostructura@gmail.com and we will fix it.

/ Related