Излезе PyTorch 2.10

PyTorch — популярен фреймворк за машинно обучение — беше обновен до версия 2.10.

Сред основните нововъведения можем да подчертаем:

• Поддръжка на AMD ROCm:

  • Введена е поддръжка на групова GEMM.
  • Подобрена е поддръжката на ROCm за Windows.
  • Добавени са нови модели GPU GFX1150/GFX1151 (серия AI 300) в списъците за поддръжка на hipblaslt.
  • Разширени функции, като поддръжка на scaled_mm v2 и AOTriton scaled_dot_product_attention.

• Поддръжка за GPU Intel:

  • Внедрени са нови API Torch XPU за Intel.
  • Поддръжка на допълнителни оператори ATen.
  • Оптимизация на производителността за GPU Intel.

• Поддръжка на NVIDIA CUDA:

  • Разширени възможности за писане на шаблонни ядра.
  • Подобрена поддръжка на CUDA 13.

• Поддръжка на Python 3.14 за torch.compile(), както и експериментална поддръжка на компилации на Python 3.14 без глобално блокиране на интерпретатора (free-threaded).

• Намаляване на разходите при стартиране на ядра чрез хоризонтално сливане на комбинирани ядра в Torch Inductor.

Пълен списък на промените е наличен на github.

Готови версии, компилирани за различни GPU, ОС и езици (Python/C++/Java), са посочени на сайта Pytorch

Източник: linux.org.ru