Google ouvre TorchTPU pour exécuter PyTorch sur ses accélérateurs
Google a annoncé le 20 octobre 2026 l’ouverture de TorchTPU, un projet destiné à faciliter l’utilisation de PyTorch sur ses accélérateurs de calcul TPU.
PyTorch constitue l’un des principaux environnements logiciels employés pour développer et entraîner des modèles d’intelligence artificielle. Son utilisation a historiquement été particulièrement associée aux accélérateurs graphiques GPU.
Avec TorchTPU, Google cherche à permettre aux développeurs d’employer plus directement leurs outils PyTorch habituels sur les infrastructures TPU.
Le projet a été développé avec des contributions de l’écosystème, notamment de Meta et de Hugging Face.
L’intérêt principal réside dans une réduction des adaptations logicielles nécessaires pour changer d’infrastructure matérielle.
Google présente des possibilités de débogage, de compilation et d’intégration avec des bibliothèques utilisées pour les grands modèles, dont vLLM, SGLang et Ray.
L’entreprise communique également des indicateurs de performances sur certaines configurations TPU. Ces résultats doivent être replacés dans leur contexte : ils concernent des matériels, modèles et méthodes de mesure spécifiques.
Une compatibilité facilitée ne signifie pas automatiquement qu’un TPU sera plus rapide ou moins coûteux qu’un GPU dans toutes les situations.
L’ouverture du code pourrait renforcer la diversité des infrastructures disponibles pour les chercheurs et entreprises.
Sources :
