I'd love to see training working on arm + mps macs. Hopefully someone knowledeable chimes in, that it's actually possible. I tried running some mpx supported c

I'd love to see training working on arm + mps macs. Hopefully someone knowledeable chimes in, that it's actually possible. I tried running some mpx supported conda/pytorch inference (whisper to be precise) on a workplace issued M1 mac and it did fall back to cpu. I encountered: - `NotImplementedError: The operator 'aten::_fft_r2c' is not currently implemented for the MPS device` - `TypeError: Trying to convert ComplexFloat to the MPS backend but it does not have support for that dtype`. EDIT: just read up it might have more to do with audio processing. Fingers crossed for better mac support in ML in general! Competition is a great thing