Support for MPS? - Githubissues

pytorch / torchtune

PyTorch native finetuning library

https://pytorch.org/torchtune/main/

BSD 3-Clause "New" or "Revised" License

4.17k stars 400 forks source link

Closed blackp2 closed 2 weeks ago

blackp2 commented 2 months ago

I was wondering if torchtune could support apple silicon. It looks like all the examples are for CUDA.

joecummings commented 2 months ago

Hi @blackp2!

This is currently under development - feel free to follow along with that PR #790

blackp2 commented 2 months ago

Awesome! Thank you.