TorchTPU replaces PyTorch/XLA and is released under the Apache 2.0 open-source license across four GitHub repositories.

From the articleGoogle Open-Sources TorchTPU, a Native PyTorch Backend for Cloud TPUsAlso in 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, Google open-sources TorchTPU to bring native PyTorch support to its TPUs, Google Open-Sources TorchTPU, a Native PyTorch Backend for Cloud TPUs