pips
pips copied to clipboard
Added support for Apple silicon
This PR adds a check for torch MPS support when CUDA is not present and uses it if it's there.
It's tested on Macbook Pro with M1 Max running Ventura 13.5
@aharley Hi Adam, not sure if it makes sense to merge into your repo. but if so, please feel free. Otherwise I'll maintain my fork.
Keen to see PIPS++!