TorchTPU: Running PyTorch Natively on TPUs at Google Scale(developers.googleblog.com)

https://developers.googleblog.com/torchtpu-running-pytorch-natively-on-tpus-at-google-scale/