Amazon представила TorchNeuron — инструмент, который, по заявлению компании, позволяет запускать код PyTorch на её чипах Trainium без переписывания: разработчику достаточно изменить настройку выбора чипа. Как сообщает IT Home, выпуск инструмента с открытым исходным кодом намечен на 20 октября, во время конференции PyTorch в Сан-Хосе.
При переходе на другой чип разработчикам обычно приходится учитывать особенности его программной платформы. Amazon утверждает, что с TorchNeuron инструменты, которыми уже пользуются для проверки, отладки и масштабирования моделей на PyTorch, смогут работать с Trainium без изменений.
Инженеры Amazon запустили один и тот же сценарий обучения ИИ-модели на обычных чипах и на Trainium, изменив только параметр выбора чипа. По данным компании, модель в обоих случаях усвоила одно и то же, а скорость обучения совпала. Amazon также работает с сообществом PyTorch над устранением в самом фреймворке кода, привязывающего его к конкретным чипам.
Источник: IT Home.
