Google open-sources TorchTPU, joining Amazon's TorchNeuron in challenging Nvidia's hold on PyTorch AI workloads
On Tuesday Google released TorchTPU as an open-source project. It lets PyTorch AI applications run on Google's Tensor Processing Units (TPUs) and Google Cloud through a configuration change, with no application rewrite. A day earlier Amazon announced TorchNeuron, which ports PyTorch apps to its Trainium chip and AWS. Amazon is also submit…