TorchTPU: Running PyTorch Natively on TPUs at Google Scale

developers.googleblog.com