Amazon ouvre TorchNeuron pour exécuter PyTorch sur ses puces Trainium

Amazon a annoncé sur son blog officiel, le 19 (heure locale), la sortie de TorchNeuron, un outil qui permet d'exécuter du code PyTorch directement sur ses puces d'IA Trainium sans modification. Selon digitaltoday.co.kr, il suffit aux développeurs de changer un seul paramètre pour transférer leur code PyTorch vers la nouvelle puce.

Les développeurs peuvent continuer à utiliser leurs outils existants de test, de débogage et de mise à l'échelle. TorchNeuron prend en charge les puces Trainium2 et Trainium3.

Amazon publiera TorchNeuron en open source à l'occasion de la PyTorch Conference North America, le 20 octobre 2026. Hugging Face propose par ailleurs un support par défaut de Trainium dans ses outils de développement d'IA, et la société de voix artificielle Cartesia exécute des charges d'inférence sur Trainium avec presque aucune modification de son code PyTorch. Le River Neuroscience Institute, qui étudie des traitements contre des troubles neuropsychiatriques comme la schizophrénie, utilise TorchNeuron dans le développement de médicaments.

Réduire le temps des développeurs

Jusqu'à présent, les développeurs qui utilisaient PyTorch devaient réécrire et vérifier leur code pendant plusieurs mois pour le porter sur une nouvelle puce. Amazon souligne que, pour de nombreuses entreprises, y compris les start-up, le coût le plus important est le temps des développeurs plutôt que les puces. L'entreprise travaille aussi avec la communauté PyTorch sur des améliorations du code visant à éviter que PyTorch ne soit lié à des puces spécifiques.

Amazon met en avant que faciliter le changement de puce par les développeurs accroîtrait la concurrence entre fabricants de puces et réduirait les coûts de développement.