Getting Pytorch to play nicely will pay dividends for not only M1 Max but for M1 Ultra, 2x M1 Ultra (Mac Pro), M2, M3, M4 and so on. It'll help Apple sell more Mac Studios and Pros. Basic testing can be done on M laptops.
Apple Silicon could potentially become an excellent valued option for ML because of its unified memory architecture. At some point in the next few years, you might be able to get 512GB of unified memory via the Mac Pro. You can't get this much vRAM today unless you're renting super expensive clusters or spend millions on local hardware.
And who knows? In the future, Apple could release Apple Silicon Cloud:
Mark Gurman is saying that Apple is working on a 40-core SoC for the Mac Pro for 2022. You're Tim Cook, sitting in his nice office, looking at how much money you just spent to make this giant SoC for a relatively small market. In fact, you have to do this every year or every two years to keep...
forums.macrumors.com