TorchTPU Google: PyTorch nativo open-source para TPUs e mais escolha além das GPUs NVIDIA

Na PyTorch Conference 2026, a Google anunciou que o TorchTPU, seu stack nativo de PyTorch para Cloud TPUs, está agora disponível como open-source no GitHub. Com isso, as TPUs passam a ser uma opção nativa ao lado de GPUs e CPUs no ecossistema PyTorch.

O TorchTPU Google foi desenvolvido em parceria com o time de PyTorch da Meta e permite executar cargas de trabalho PyTorch existentes em TPUs com alterações mínimas no código, usando as APIs familiares. É possível depurar em modo eager e depois migrar para desempenho de produção via torch.compile.

Por que isso importa

A maior parte do desenvolvimento de IA hoje parte do PyTorch e, por padrão, está otimizada para GPUs NVIDIA e CUDA. Tanto a Amazon (com o TorchNeuron para chips Trainium) quanto a Google estão agora oferecendo caminhos nativos para seus aceleradores próprios, reduzindo o lock-in de software.

Jana van Greunen, Diretora de Engenharia de PyTorch na Meta, destacou que a empresa precisa de diversidade de hardware e que o TorchTPU permite implantar cargas de produção (LLMs, anúncios e sistemas de recomendação) em TPUs da Google.

Integrações do ecossistema

  • Hugging Face: modelos open-weight do Hub rodam nativamente; no Transformers v5 basta device_map="tpu".
  • vLLM e SGLang: backends nativos para serving de alta vazão e cargas agentic.
  • TorchTitan: treino distribuído validado em até 8.192 chips Ironwood (TPU v7x), com até 52% de Model Flops Utilization em Qwen 3 8B.
  • VeRL, Ray e LiteRT: suporte a RL, orquestração heterogênea e inferência on-device.

A Google também lançou em preview o TPU agent suite, que ajuda a automatizar onboarding, diagnósticos e tuning de desempenho.

O movimento chega um dia depois do anúncio do TorchNeuron da Amazon e reforça a tendência de ferramentas abertas que tornam o PyTorch mais portável entre diferentes aceleradores.

Fontes

Transparência: Este conteúdo foi criado, editado ou revisado com auxílio de inteligência artificial. As informações foram cruzadas com publicações públicas no X e fontes disponíveis na internet. Consulte as fontes originais para verificar o contexto completo.

Por GeekikiBot