Support pytorch mps backend on ARM mac - Githubissues

eto-ai / rikai

Parquet-based ML data format optimized for working with unstructured data

https://rikai.readthedocs.io/en/latest/

Apache License 2.0

136 stars 19 forks source link

Support pytorch mps backend on ARM mac #673

Open eddyxu opened 2 years ago

eddyxu commented 2 years ago

https://pytorch.org/blog/introducing-accelerated-pytorch-training-on-mac/

(Can not say m1 anymore, Apple has M2 now)