IT之家 10月20日消息，当地时间10月19日，亚马逊宣布推出开源工具TorchNeuron。借助该工具，开发者编写的PyTorch代码可在亚马逊自研的Trainium芯片上本地运行，只需更改一次配置，无需重写代码。
IT之家获悉, TorchNeuron将于当地时间10月20日在加利福尼亚州圣何塞举行的北美PyTorch大会上以开源许可形式正式发布。
PyTorch — это открытая платформа, разрабатываемая Meta, которая используется большинством разработчиков искусственного интеллекта. NVIDIA в течение более десяти лет разрабатывала инструменты CUDA, позволяющие эффективно запускать код PyTorch на чипах NVIDIA.
Согласно информации, инженеры Amazon запустили один и тот же скрипт обучения ИИ на традиционных чипах и на чипах Trainium. Единственное различие заключалось только в параметрах настройки, указывающих используемый чип в коде. Результаты показали, что модель ИИ одинаково хорошо обучилась в обоих окружениях и продемонстрировала одинаковую скорость обучения. Это означает, что инструменты, которые разработчики использовали для тестирования, отладки и масштабирования моделей ИИ, можно использовать на Trainium без каких-либо изменений.
Amazon заявляет, что TorchNeuron привносит в индустрию искусственного интеллекта долгожданный ключевой элемент: настоящую возможность выбора. Когда разработчики могут свободно переключаться между различными чипами без дополнительных затрат, производители чипов вынуждены конкурировать, предлагая более конкурентоспособные продукты, чтобы завоевать симпатии разработчиков; в свою очередь, разработчики получают выгоду в виде снижения затрат и ускорения циклов итераций.
Кроме того, Amazon активно участвует в разработке PyTorch вверх по потоку, сотрудничая с сообществом PyTorch для удаления кода, привязывающего фреймворк к конкретным чипам. Их цель — сделать PyTorch более «независимым от аппаратного обеспечения» для всего сообщества разработчиков, а не только для пользователей Trainium.