PyTorch — un popular marco para el aprendizaje automático — se ha actualizado a la versión 2.10.
Entre las principales novedades se destacan:
Soporte para AMD ROCm:
- Se ha introducido soporte para GEMM agrupado.
- Mejora del soporte de ROCm para Windows.
- Nuevos modelos de GPU GFX1150/GFX1151 (serie AI 300) añadidos a la lista de soporte hipblaslt.
- Se han ampliado funcionalidades como el soporte para scaled_mm v2 y AOTriton scaled_dot_product_attention.
Soporte para GPU Intel:
- Se han implementado nuevas API Torch XPU para Intel.
- Soporte para operadores adicionales de ATen.
- Optimización del rendimiento para GPU Intel.
Soporte para NVIDIA CUDA:
- Ampliadas las capacidades para escribir núcleos templados.
- Mejora del soporte para CUDA 13.
Soporte para Python 3.14 en torch.compile(), así como soporte experimental para la construcción de Python 3.14 sin bloqueo global del intérprete (free-threaded).
Reducción de la sobrecarga al iniciar núcleos gracias a la fusión horizontal de núcleos combo en Torch Inductor.
La lista completa de cambios está disponible en github.
Las versiones listas, compiladas para diferentes GPU, sistemas operativos y lenguajes (Python/C++/Java) se indican en el sitio web de Pytorch.
Fuente: linux.org.ru
