search

SJ001 / AI-Feynman

MIT License
607 stars 186 forks source link

Add PyTorch back-end support for 'mps' #66

Closed dbl001 closed 2 years ago

dbl001 commented 2 years ago

PyTorch has added GPU support for Apple Metal ('mps'). Please extend AI-Feynman's GPU support (currently 'cuda') to include 'mps'.

Thanks for the excellent software!