himalaya icon indicating copy to clipboard operation
himalaya copied to clipboard

Published 20 hours ago • verified publisher icon gallantlab

WIP torch MPS backend

Open mvdoc opened this issue 2 years ago • 0 comments

As described in #42, torch's MPS backend is not fully ready yet. I'm adding this draft PR just to keep track of what I've tried so far. There were some workarounds required due to MPS supporting only float32.

mvdoc avatar Feb 15 '23 04:02 mvdoc