PyTorch est une bibliothèque connue du monde du machine learning. Open source, elle est basée sur Torch, bibliothèque éditée par Meta. Contrairement à cette dernière, PyTorch s’exploite en Python.

L’équipe chargée du développement a présenté hier soir une préversion de la bibliothèque optimisée pour Apple Silicon, plus précisément conçue pour exploiter la partie GPU de ces puces via Metal, avec le concours d’Apple.

Bien qu’il ne s’agisse que d’une préversion, les résultats sont à l’avenant. Sur un M1 Ultra, les performances sont multipliées par 5 à 20 en fonction des tâches.

Tous les développeurs peuvent utiliser cette version, à condition d’avoir au moins macOS 12.3, un Mac M1 (et ses variantes) et la mouture arm64 de Python.