Google lança TorchTPU para rodar PyTorch em chips TPU
25 de outubro de 2026 · 1 min de leitura
Google lançou o TorchTPU, um backend da biblioteca PyTorch que permite a execução nativa e otimizada em chips TPU. Esta iniciativa é semelhante ao lançamento do TorchNeuron pela Amazon, que compila código PyTorch para execução em chips Trainium.
O TorchTPU é resultado de uma colaboração entre Google e a equipe do PyTorch da Meta, visando expandir o uso do PyTorch em novos chips além dos tradicionais CPU e GPU.
Além disso, Google estabeleceu uma parceria com a Hugging Face para testar a execução de diversos modelos abertos no TorchTPU, garantindo que a performance seja realmente eficaz. Também foram realizados testes com bibliotecas renomadas como vLLM, SGLang, llm-d, TorchTitan, VeRL, Ray e LiteRT para avaliar a funcionalidade no TPU.
O desenvolvimento do TorchTPU começou em 2025 e foi oficialmente lançado como um projeto de código aberto no GitHub durante a PyTorch Conference 2026, realizada nesta semana.