Излезе 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