Google et Amazon élargissent PyTorch aux TPUs et à Trainium

1 min de lecture

Google et Amazon élargissent PyTorch aux TPUs et à Trainium

TorchTPU de Google
Google a rendu public TorchTPU, un projet open source destiné à permettre le portage du framework PyTorch sur ses unités de traitement Tensor (TPU). Cette initiative vise à faciliter l’utilisation de PyTorch dans les environnements Google Cloud, où les développeurs pourront exécuter leurs modèles sur l’infrastructure TPU de Google.

TorchNeuron d'Amazon
Amazon a lancé TorchNeuron, un outil conçu pour optimiser les modèles PyTorch sur sa puce d’accélération Trainium. En parallèle, Amazon contribue à l’ouverture de PyTorch à d’autres matériels, élargissant ainsi la compatibilité du framework au-delà des GPU traditionnels.

Contexte du marché
Historiquement, PyTorch a été fortement associé aux GPU de Nvidia, qui restent les principaux fournisseurs de matériel d’accélération pour l’apprentissage profond. Les récentes initiatives de Google et d’Amazon marquent une diversification des options matérielles disponibles pour les utilisateurs de PyTorch.

Source : Brief IA