TorchTPU 现已开源：在 TPU 上运行原生 PyTorch
TorchTPU is now open source: Run native PyTorch on TPUs
Today at PyTorch Conference 2026, we announced a range of new products and integrations designed to help you run PyTorch effectively on Google Cloud or in your own infrastructure. Notably, TorchTPU, our native PyTorch stack for Google TPUs, is now available as open source on GitHub. With TorchTPU, TPUs join GPUs and CPUs as a native option for PyTorch. You can now train, serve, and deploy models on TPUs using the tools you already rely on, including vLLM, SGLang, llm-d, TorchTitan, Ray, TorchRec, Hugging Face, LiteRT. Ultimately, this provides our customers choice across a broad range of…
来源 google.com
报道
- TorchTPU 现已开源：在 TPU 上运行原生 PyTorch TorchTPU is now open source: Run native PyTorch on TPUs google.com · · 首发