PyTorch 2.10 har släppts

PyTorch – en populär ramverk för maskininlärning har uppdaterats till version 2.10.

Bland de viktigaste nyheterna kan nämnas:

• Stöd för AMD ROCm:

  • Stöd för grupperad GEMM har införts.
  • Förbättrat stöd för ROCm på Windows.
  • Nya GPU-modeller GFX1150/GFX1151 (AI 300-serien) har lagts till i hipblaslt:s stödlista.
  • Utökade funktioner, såsom stöd för scaled_mm v2 och AOTriton scaled_dot_product_attention.

• Stöd för Intel GPU:

  • Nya API:er för Torch XPU för Intel har implementerats.
  • Stöd för ytterligare ATen-operatörer.
  • Prestandaoptimering för Intel GPU.

• Stöd för NVIDIA CUDA:

  • Utökade möjligheter för att skriva mallkärnor.
  • Förbättrat stöd för CUDA 13.

• Stöd för Python 3.14 för torch.compile(), samt experimentellt stöd för att bygga Python 3.14 utan global låsning av tolken (free-threaded).

• Minskade overhead-kostnader för att köra kärnor tack vare horisontell sammanslagning av combo-kärnor i Torch Inductor.

En fullständig lista över ändringar finns på github.

Färdiga alternativ, kompilerade för olika GPU:er, operativsystem och språk (Python/C++/Java) anges på Pytorch:s webbplats

Källa: linux.org.ru