Amazon ouvre TorchNeuron en Apache 2.0 pour faire tourner PyTorch sur Trainium sans réécriture
Résumé
Amazon dévoile TorchNeuron, backend PyTorch natif pour les puces Trainium2 et Trainium3, publié en open source Apache 2.0 à la PyTorch Conference North America du 20 octobre. Un simple `.to('neuron')` fait basculer le code existant, torch.compile et les API distribuées (FSDP2, tenseur, pipeline) sans réécriture. Deepgram rapporte +60 % de secondes audio traitées par dollar contre son baseline GPU, Hugging Face Transformers v5 et Diffusers passent nativement, visant à casser l'avantage logiciel CUDA de Nvidia.
Article original publié par pytorch.org
Lire l'article original →Titre original : Amazon ouvre TorchNeuron en Apache 2.0 pour faire tourner PyTorch sur Trainium sans réécriture