В блоге PyTorch команда разработчиков рассказала, что предварительная версия фреймворка машинного обучения получила поддержку шейдеров Apple Metal Performance Shaders (MPS) в качестве серверной архитектуры вычислений.
Разработчики отметили, что устройства на базе Apple Silicon оснащены унифицированной системой памяти, предоставляющей GPU прямой доступ ко всей памяти сразу. Такой подход делает Mac удобной площадкой для обучения больших сетей и моделей. Также команда PyTorch утверждает, что обновление поможет разработчикам экономить на облачных решениях для обучения.
Нововведение доступно в предварительной сборке PyTorch Preview (Nightly). Для установки необходима версия macOS 12.3 и выше. Подробная инструкция доступна на сайте PyTorch и Apple Developer.
