Подразделение Annapurna Labs компании Amazon представило TorchNeuron — программный бэкенд с открытым кодом для запуска PyTorch на чипах Trainium. Инструмент позволяет перенести код с оборудования Nvidia на процессоры Amazon, заменив одну строку: .to(cuda) на .to(neuron).
Публичная бета-версия была представлена 20 октября 2026 года на конференции PyTorch, анонс состоялся днем ранее. TorchNeuron распространяется по лицензии Apache 2.0 и поддерживает чипы поколений Trainium2 и Trainium3, позволяя им работать в PyTorch как нативные устройства.
В тестировании участвовало более 30 компаний, включая Cartesia, Deepgram и Lieber Institute for Brain Development. Ранние пользователи сообщают о снижении затрат при переходе на новую инфраструктуру, однако Amazon не раскрывает точные показатели экономии.
Подразделение Amazon по разработке собственных чипов достигло годового темпа выручки в $25 миллиардов, компания прогнозирует рост до $50 миллиардов. Текущая бета-версия TorchNeuron пропускает около 22 000 тестов PyTorch из-за отсутствия поддержки ряда функций, включая динамические входные формы.
