Hi, this is duplicate of #382, where we have various reports that it either fails to run some operations or performance that's not very promising. Looks like it's doing better on pytorch 2.x; I'll plan to merge #382 after confirming that MPS is works reliably and fast enough.

Hi @jongwook, this pull request may pass the failing operations. PyTorch should be installed with MPS compatibility for torch.backends.mps.is_available() to work. Otherwise, it would throw out an error. Wrapping it around try and catch statement would fix this issue.