Hm. Tempted to try pytorch on my Mac for this. I have an AS chip rather than a Nvidia GPU.