PyTorch 2.10 er ute

PyTorch — en populær ramme for maskinlæring — har blitt oppdatert til versjon 2.10.

Blant de viktigste nyhetene kan vi nevne:

• Støtte for AMD ROCm:

  • Støtte for gruppert GEMM er introdusert.
  • Forbedret støtte for ROCm på Windows.
  • Nye GPU-modeller GFX1150/GFX1151 (AI 300-serien) er lagt til i støttelisten for hipblaslt.
  • Utvidede funksjonaliteter som støtte for scaled_mm v2 og AOTriton scaled_dot_product_attention.

• Støtte for GPU-er fra Intel:

  • Nye API-er for Torch XPU for Intel er implementert.
  • Støtte for flere ATen-operatører.
  • Ytelsesoptimalisering for GPU-er fra Intel.

• Støtte for NVIDIA CUDA:

  • Utvidede funksjoner for skriving av maler.
  • Forbedret støtte for CUDA 13.

• Støtte for Python 3.14 for torch.compile(), samt eksperimentell støtte for bygging av Python 3.14 uten global låsing av tolken (free-threaded).

• Reduksjon av overhead ved oppstart av kjerner takket være horisontal sammenslåing av kombinasjonskjerner i Torch Inductor.

Den fullstendige listen over endringer er tilgjengelig på github.

Ferdige versjoner, kompilerte for forskjellige GPU-er, OS-er og språk (Python/C++/Java) er oppgitt på Pytorch-nettsiden

Kilde: linux.org.ru