dbl001 opened this issue 2 years ago · 0 comments
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!