TorchTPU: Executando PyTorch Nativamente em TPUs na Escala do Google

TorchTPU é uma iniciativa do Google que permite a execução nativa e eficiente de modelos PyTorch em hardware TPU, priorizando usabilidade e alta performance. O sistema suporta treinamento distribuído complexo. Para 2026, os planos incluem acesso a um repositório público e melhor suporte para dynamic shapes.