Announcement
Google open-sources TorchTPU for native PyTorch on TPUs
Google Cloud says the stack, co-developed with Meta’s PyTorch team, runs existing PyTorch workloads on TPUs with minimal code changes.
TLDR
Google Cloud announced that TorchTPU, its native PyTorch stack for Google TPUs, is available as open source on GitHub. It says developers can use familiar PyTorch tools to train and serve models on TPUs, with integrations including vLLM, SGLang, TorchTitan and Ray.
Combined views
1.6K
1 Source, first seen ago
