Google releases open-source TorchTPU to run PyTorch on TPUs, challenging Nvidia
Google announced TorchTPU, an open-source PyTorch implementation that lets developers run existing PyTorch apps on Google’s TPUs with minimal changes; the move follows Amazon’s recent release of TorchNeuron for Trainium and accompanies efforts to decouple PyTorch from Nvidia-specific code. Google also released a TPU agent suite and updated LiteRT to assist porting and edge deployments.