Hugging Face explains running Transformers and Diffusers natively on TPUs with TorchTPU
AIHugging Face says TorchTPU adds TPU as a native PyTorch backend, so models can be sent to a TPU device without the older torch_xla path. The post shows device_map="tpu" for Transformers and pipe.to("tpu") for Diffusers, and reports that 96.5% of the TPU tests it runs pass, with 34 reported issues and 19 already fixed.
AIWhy it matters.
The post explains how TorchTPU makes TPUs a native PyTorch device and gives working code for running Transformers and Diffusers models on them.
Source details
Hugging Face Blog · huggingface.coPublished · added here
Original title: Hugging Face ❤️ TorchTPU: TPUs, Natively