Google a mis en open source TorchTPU, un back end PyTorch natif destiné à ses unités de traitement de tenseurs (TPU). Le projet est désormais hébergé dans un dépôt GitHub public au sein de l'organisation google-pytorch, où les développeurs peuvent suivre et contribuer à son développement.
Cette décision compte parce que PyTorch est le framework que la plupart des chercheurs et ingénieurs en IA privilégient, tandis que les TPU ont longtemps été étroitement liés aux piles JAX et TensorFlow de Google. Une voie PyTorch officielle vers les TPU réduit cet écart et donne aux développeurs une nouvelle raison de s'intéresser aux puces de Google.
Ce que fait TorchTPU
TorchTPU se présente comme un back end natif, ce qui signifie que les charges de travail PyTorch peuvent cibler le matériel TPU via l'intégration propre au projet plutôt que par un pont externe. Cette distinction est importante : le support natif apporte généralement des performances plus propres et moins de problèmes de compatibilité que les ports tiers.
Le dépôt est développé ouvertement au sein de l'organisation GitHub de Google. D'après la fiche du dépôt, il affiche actuellement 48 tickets ouverts et 221 pull requests ouvertes, signe d'une base de code active et encore en mouvement plutôt que d'un produit fini.
- Projet : TorchTPU, un back end PyTorch natif pour les TPU
- Hébergé sur GitHub au sein de l'organisation google-pytorch
- Le dépôt affiche 48 tickets ouverts et 221 pull requests ouvertes
- La branche principale héberge le code source pour les contributions des développeurs
