Google a annoncé TorchTPU à la PyTorch Conference 2026. Cette pile logicielle permet d’exécuter PyTorch sur ses puces TPU, pour entraîner des modèles ou les utiliser, sans passer par XLA, que Google prévoit d’archiver le 15 novembre. Le projet est publié sous licence Apache 2.0 et prend en charge des outils comme Hugging Face et vLLM. Google affirme que TorchTitan a gardé une montée en charge linéaire jusqu’à 8 129 TPU ; le texte ne cite pas de validation indépendante.
Meta participe au projet, tandis que Google cherche à rendre ses puces plus simples à utiliser avec les outils déjà employés par les équipes d’IA.
Pourquoi c'est important
Pour les équipes qui entraînent des modèles avec PyTorch, TorchTPU pourrait faciliter le passage aux TPU de Google, mais la source ne précise ni les coûts ni les performances comparées à d’autres puces.