Hugging Face launches TorchTPU for native PyTorch support on Google Cloud TPUs
Verified · 6/6 supported
Hugging Face introduced TorchTPU, a native PyTorch backend for TPUs . With TorchTPU, Transformers, Diffusers and Kernels run natively on TPUs within the Hugging Face ecosystem . TPUs are Google’s AI accelerators built for matrix math and are used by Google to train and serve models such as Gemini . Google Cloud offers TPUs in configurations ranging from a single chip to pods of thousands of chips . Previously, PyTorch on TPUs required torch_xla with lazy tensors and Optimum TPU for Hugging Face models, but those are no longer needed with TorchTPU .
Why it matters. The change lets developers run Hugging Face models on Google Cloud TPUs directly, removing the need for Optimum TPU and simplifying training and inference workflows .