Apple M1 MPS device just gives noise

[robagar]

Hi - nice work btw!

On my M1 MacBook Pro it works perfectly on the CPU, but I just get noise if I try and use the hardware accelerated MPS device.

CPU:

MPS:

I assume this is really a PyTorch/MPS issue rather than with your code, but I wonder if you can shed any light on what the problem might be?

There are no errors indicating usage of PyTorch features unimplemented for Apple hardware, and unsurprisingly using PYTORCH_ENABLE_MPS_FALLBACK=1 has no effect.