Avlaiblity  of CPU/MPS version 

Thank you for the amazing work on this repository! I am still using Intel MacBook and I could run Google Colab tutorial, so I wanted to check if this project currently offers a CPU-only version or supports the MPS accelerator (for macOS with Apple Silicon). If not, are there any plans to implement support for either of these in the future?
