I encountered the same problem as you. I couldn't find a way to use PyTorch properly on the M4 Pro chip. Unfortunately. They're beautiful machines, but not well-suited for PyTorch development. Float 64 is not supported, and some layers aren't either. Very sad.