I'm receiving the same problem (slow GPU training and NUMA node warning) on my M1 max. I'd be very happy to see a fix for this issue.