Google lance TorchTPU pour exécuter PyTorch sur TPU sans réécriture
Le projet open source TorchTPU permet aux applications PyTorch existantes de tourner sur TPU et Google Cloud via de simples changements de configuration.
Google a publié le 20 octobre 2026 le projet open source TorchTPU, qui permet aux applications PyTorch existantes de s'exécuter sur des TPU et sur Google Cloud sans réécriture du code, selon digitaltoday.co.kr. Les développeurs peuvent déplacer leurs applications vers les TPU et Google Cloud avec de simples changements de configuration.
Google a également publié une suite d'outils « TPU Agent Suite » pour faciliter la migration des applications et la correction des anomalies. La société a par ailleurs mis à jour LiteRT, son environnement d'exécution d'IA embarquée, pour une meilleure intégration avec les applications PyTorch.
Le projet TorchTPU a été développé en collaboration avec Meta, Hugging Face, Inferact (vLLM) et RadixArk (SGLang) pour assurer la compatibilité.
Un contexte de concurrence entre infrastructures
Meta a développé PyTorch en 2016, à une époque où les GPU NVIDIA étaient le matériel standard pour la recherche en IA, ce qui a influencé la conception de nombreuses fonctionnalités de PyTorch. En 2022, Meta a transféré PyTorch à la fondation Linux, organisation à but non lucratif. Le 19 octobre 2026, Amazon a de son côté publié sa propre version de PyTorch, TorchNeuron, qui déplace le code des applications PyTorch vers ses puces d'IA Trainium et AWS sans réécriture. Selon TechCrunch, Amazon soumet du code au projet PyTorch pour découpler diverses fonctionnalités des puces NVIDIA.
En bref
Qu'est-ce que TorchTPU ?
C'est un projet open source de Google qui permet aux applications PyTorch existantes de s'exécuter sur des TPU et sur Google Cloud sans réécriture du code.
Faut-il modifier le code de son application PyTorch pour utiliser TorchTPU ?
Non, seuls de simples changements de configuration sont nécessaires pour déplacer une application vers les TPU et Google Cloud.
Avec quels partenaires Google a-t-il collaboré sur TorchTPU ?
Google a collaboré avec Meta, Hugging Face, Inferact (vLLM) et RadixArk (SGLang) pour la compatibilité de TorchTPU.