Added support for Apple Metal Performance Shaders (mps) on M3

[zoharbabin]

few minor changes to add support for mps. note that ffc.py makes use of torch.fft.rfftn which is only supported in cuda, so this operation falls back to cpu in case mps was the enabled device. once the operation is finished, the work is moved back to mps.