GPU Support for Apple M1

[visuallization]

Hey there,

Thanks for all the work you have put into this library!
I was wondering if there will be support for GPU training on Apple M1?
Apparently pytorch will be officially supporting GPU accelerated training on macs.

https://pytorch.org/blog/introducing-accelerated-pytorch-training-on-mac/
https://sebastianraschka.com/blog/2022/pytorch-m1-gpu.html

Cheers

[alex-petrenko]

I will soon get my hands on one of these laptops and it'd be great to add this support.
I imagine it can be done with just a few changes in CUDA-specific code.
Let's keep this feature open to discuss any development issues. Contributions are welcome!