SJ001/AI-Feynman

Can we add support for 'mps'

Opened this issue · 0 comments

dbl001 commented

PyTorch has just added support for 'mps' GPUs.
Will this eventually be integrated into AI Feynman?