No luck needed. This is for the GPU and is already available via Metal. https://x.com/liuliu/status/1932158994698932505
Ill believe it when I see it supported in Pytorch