Can we add support for 'mps'

SJ001 / AI-Feynman

MIT License

607 stars 186 forks source link

Open dbl001 opened 2 years ago

dbl001 commented 2 years ago

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