Thank you! I wanted to write something that simply "just works".

Currently, I'm not aware of a simple way to use Pytorch in M1, and in any case the alternatives if explored are based on Rosseta2 and they are not using MLCompute at all. Hopefully, things will evolve soon and I will try to write something like this the moment it a simple alternative is available.