Google abre o código do TorchTPU, trazendo PyTorch nativo para as suas TPUs
A Google tornou open source, a 20 de outubro, o TorchTPU, a sua stack de PyTorch nativo para TPUs, desenvolvida em conjunto com a equipa de PyTorch da Meta e anunciada na PyTorch Conference 2026.
O TorchTPU permite correr cargas de trabalho PyTorch já existentes em TPUs com alterações mínimas de código, cobrindo treino, serving e implantação. É distribuído sob licença Apache 2.0, no repositório 'torch_tpu' da organização google-pytorch, com o pacote torch-tpu 0.2.0 já disponível no PyPI desde 19 de outubro.
Tecnicamente, oferece três modos 'eager' (Debug, Strict e Fused) com cache de compilação partilhado — o modo Fused é entre 50% e mais de 100% mais rápido do que o Strict, segundo dados internos de abril de 2026 — além de compilação via torch.compile, suporte a kernels personalizados em Pallas/JAX e a técnicas distribuídas como FSDPv2 e DTensor. A Google diz ter validado a stack até 8.000 chips Ironwood (TPU v7x), com 52% de utilização de FLOPs no modelo Qwen 3 8B.
Entre as integrações já anunciadas estão o Hugging Face Transformers v5, o vLLM, o SGLang e o llm-d 1.0 (ver item seguinte). Estes números de desempenho são divulgados pela própria Google e seus parceiros, sem verificação independente encontrada até ao momento, e o anúncio não compara diretamente o desempenho com a stack equivalente da Nvidia.
Perguntas frequentes
O que é o TorchTPU da Google?
É a stack de PyTorch nativo da Google para TPUs, agora disponível em código aberto sob licença Apache 2.0, permitindo correr cargas de trabalho PyTorch em TPUs com alterações mínimas de código.
Fontes
- Unite.AI www.unite.ai
Resumo editorial em português feito com apoio de inteligência artificial a partir das fontes indicadas. Como fazemos o Boletim IA.