Add --device mps to support Apple's MPS backend

Dear devs,

Would you consider adding support for MPS (Metal Performance Shaders) which is available as an alternative non CPU backend on Apple Silicon devices?

My own dirty experiments show that there are no intrusive changes needed, "mps" needs to be exposed to commandline interface e.g. `--device` flag.

Thank You!