Google Open-Sources TorchTPU to Challenge NVIDIA's Dominance in the AI Ecosystem
TorchTPU, jointly developed by Google Cloud and Meta's PyTorch team, has been open-sourced on GitHub. TPU is now a native PyTorch option alongside GPU and CPU, allowing existing code to run with virtually no changes. Frameworks like vLLM, SGLang, llm-d, and TorchTitan have already integrated it, and Hugging Face Transformers v5 can deploy models directly to TPU.
Google Open Sources TorchTPU to Challenge NVIDIA's Dominance in the AI Ecosystem
Google Cloud has announced that TorchTPU, developed in collaboration with Meta's PyTorch team, is now open source on GitHub. This makes TPU a native option for PyTorch alongside GPU and CPU, allowing existing code to run with virtually no modifications.
Frameworks such as vLLM, SGLang, llm-d, and TorchTitan have already been integrated, and Hugging Face Transformers v5 can deploy models directly to TPU.
Google Cloud: TorchTPU is now open source
