AWS lance TorchNeuron pour exécuter PyTorch sur les puces Trainium
L'outil d'Amazon Web Services promet de faire tourner les applications PyTorch sur Trainium en modifiant une seule ligne de configuration.
Amazon Web Services a publié TorchNeuron, un outil permettant aux applications d'IA écrites avec PyTorch de fonctionner sur les puces Trainium d'Amazon en changeant une seule ligne de configuration, le 19 octobre 2026.
Selon Amazon, TorchNeuron ne nécessite aucune réécriture du code et les développeurs familiers de PyTorch peuvent l'utiliser directement. AWS indique également que Trainium prend en charge l'entraînement et l'inférence sur une même puce et qu'il est intégré à son infrastructure cloud.
Amazon affirme par ailleurs contribuer en amont au projet principal PyTorch afin de supprimer le code qui lie le framework à des puces spécifiques. La société estime qu'environ 85 % des applications d'IA sont écrites avec PyTorch.
Un framework longtemps lié à Nvidia
PyTorch fait partie de la Linux Foundation depuis 2022 ; il avait été développé à l'origine par Facebook. Nvidia a passé des années à faire en sorte que son logiciel CUDA prenne en charge les applications d'IA écrites avec PyTorch.
Colin Brace occupe le poste de vice-président Technology Software chez AWS Annapurna Labs.
En bref
Qu'est-ce que TorchNeuron ?
Un outil publié par AWS qui permet aux applications d'IA écrites avec PyTorch de tourner sur les puces Trainium d'Amazon en modifiant une seule ligne de configuration.
Faut-il réécrire le code de son application ?
Non, selon Amazon : TorchNeuron ne nécessite aucune réécriture du code et les développeurs familiers de PyTorch peuvent l'utiliser directement.