Hello, I wanted to check whether PyTorch runs on Apple MacBook Air M4 Chip 2025, with 10 core CPU, 10 core GPU, 16 core neural engine. Thank you.
1 Like
Hey, the Metal backend for torch is currently getting some attention from a few contributors.
The short answer is yes, it will run, but some operators are missing support, or have incomplete support, both in eager and under torch.compile. The CPU device should be fully featured on apple hardware.
This tracking issue has links to different reference materials which are updated regularly indicating the prioritization under which we are working through things as well as the current status of stuff we are tracking.
I would encourage you to try out what you are looking to do and see how it works!
2 Likes