Google and Meta open-source TorchTPU for native PyTorch on TPUs

infoq.com2 min read

Google and Meta have open-sourced TorchTPU, a project that provides a native PyTorch backend for Google TPUs with minimal code modifications. The approach prioritizes eager execution and supports distributed computing, hybrid compilation modes, and extensive testing across thousands of Hugging Face models. Google reports strong performance for inference and training workloads on TPUv7 hardware. Open article →