AWS анонсирует TorchNeuron: встроенную поддержку PyTorch для AWS Trainium (публичная бета-версия)

Проведено: 20 окт. 2026 г.

Сегодня AWS запустила публичную бета-версию TorchNeuron, встроенного бэкенда PyTorch с открытым исходным кодом для AWS Trainium. Trainium – это семейство чипов ИИ, разработанных для обеспечения высокой производительности и экономичности рабочих нагрузок глубокого обучения, обучения ИИ и логического вывода. TorchNeuron позволяет ученым и исследователям в области ИИ экспериментировать и создавать модели с PyTorch прямо на Trainium.

С помощью TorchNeuron разработчики могут запустить свой код PyTorch на Trainium, просто переключившись на .to('neuron'), сохранив при этом полную совместимость PyTorch с немедленным выполнением, torch.compile и стандартными API PyTorch. Кроме того, TorchNeuron интегрируется с Нейронным интерфейсом Kernel (NKI), что позволяет настраивать операторы и оптимизировать производительность для оборудования Trainium.

TorchNeuron доступен в виде открытого исходного кода в репозитории TorchNeuron на GitHub для инстансов Amazon EC2 Trainium2 и инстансов Amazon EC2 Trainium3.

Подробнее о TorchNeuron и ускорителях AWS Trainium см. на странице AWS Neuron.