TorchTPU registers Google TPUs as standard hardware devices using PyTorch's PrivateUse1 interface, enabling eager execution without source code modifications or wrappers.

From the articleGoogle open-sources TorchTPU to bring native PyTorch support to its TPUsAlso in Google Open-Sources TorchTPU, a Native PyTorch Backend for Cloud TPUs, Google Open-Sources TorchTPU, a Native PyTorch Backend for Cloud TPUs, Switch Your PyTorch Model From Nvidia to Google TPUs With One Line: TorchTPU Is Now Open Source, Switch Your PyTorch Model From Nvidia to Google TPUs With One Line: TorchTPU Is Now Open Source