Aller au contenu
Édition du soir · N° 558 · prochaine à 9 h
Compte
Connexion

Au clavier : 1 Aujourd'hui, 2 Rechercher, 3 Modèles et outils, 4 Acteurs ; J et K d'un article à l'autre, S pour sauvegarder.

Fellow
La veille de l'intelligence artificielle, en français
Rechercher⌘K
OUTILS · Hugging Face
OUTILVerdict : notable

Hugging Face et TorchTPU : un support natif des TPU dans PyTorch

Avec TorchTPU, backend PyTorch natif, le TPU devient un device comme un autre : Transformers, Diffusers et Kernels tournent directement dessus, sans passer par torch_xla.

Hugging Face Blog·20 octobre 2026·

Sources : Hugging Face Blog (source primaire), google-pytorch sur GitHub

Lu et jugé par Fellow · impact notable

TorchTPU est disponible via pip et rend les TPU utilisables nativement dans Transformers et Diffusers, avec des gains mesurés (jusqu'à 21,5× en décodage MoE), mais l'avancée reste incrémentale pour l'écosystème.

Verdict rendu par une IA selon une grille publique

Discuter de ce verdict

Hugging Face et TorchTPU : un support natif des TPU dans PyTorch
Image · Source originale
TorchTPU ajoute le TPU comme backend natif de PyTorch, avec des kernels Pallas optimisés, le support de l'entraînement distribué et une intégration de torch.compile. Transformers, Diffusers et Kernels fonctionnent nativement sur TPU, par exemple via device_map="tpu". Optimum TPU n'est plus nécessaire pour exécuter les modèles Hugging Face avec Transformers et Diffusers. L'installation se fait par pip install torch-tpu.

Traduit et résumé par IA à partir de l'article de Hugging Face Blog ↗ · Comment Fellow travaille

Fiche TorchTPU : éditeur, prix et articles

OUVRIR LA SOURCE ↗
#tpu#pytorch#hugging-face#transformers#infra

Une question sur cet article ?

Fellow répond à partir de ce qu'il a publié et cite chaque source. Réponses rédigées par une IA.

À lire aussi