TorchTPU: Running PyTorch Natively on TPUs at Google Scale (10 minute read)

Google's Tensor Processing Units (TPUs) are foundational to the company's supercomputing infrastructure. The company's custom ASICs power