Add support for Apple silicon MPS#721

Lasidar wants to merge 3 commits intoostris:mainostris/ai-toolkit:mainfrom

Lasidar:feature/mps-support-rebaseLasidar/ai-toolkit:feature/mps-support-rebaseCopy head branch name to clipboard