Amazon представила TorchNeuron — инструмент для запуска PyTorch на чипах Trainium
Amazon представила TorchNeuron — открытый программный модуль для переноса моделей машинного обучения на фирменные ускорители Trainium 2 и Trainium 3.
Разработчикам при этом доступны привычные инструменты для обучения, запуска и отладки нейросетей.TorchNeuron подключает чипы Amazon к PyTorch с помощью стандартного протокола расширений фреймворка.
habr.com