MPS support #152
Description
Is your feature request related to a problem? Please describe.
pythae supports CUDA and CPU, but no MPS
Describe the solution you'd like
I would like pythae to use MPS if no CUDA is available
Additional context
fixed (in a rudimentary fashion) by my pull request #151