PyTorch 2.15 прокачал MPS: обучение на Mac стало быстрее

PyTorch 2.15 принёс заметные улучшения для пользователей macOS с чипами M1/M2/M3. Разработчики усилили бэкенд MPS — теперь он поддерживает больше операторов, включая некоторые сложные слои для трансформеров и сверточных сетей. Это значит, что обучение и инференс на Apple Silicon стало быстрее и стабильнее без дополнительных настроек.

Ключевое — оптимизация работы с памятью и улучшенная поддержка смешанной точности (bfloat16). Также в релизе исправлены баги и добавлены новые функции для работы с графами вычислений. Если вы гоняете модели на Mac, обновление стоит поставить — прирост производительности особенно заметен на больших батчах и длинных последовательностях.