Add MPS (Apple Silicon) support via jax-mps#69

vlordier wants to merge 1 commit into

learnsyslab:mainlearnsyslab/crazyflow:mainfrom

vlordier:feat/mps-supportvlordier/crazyflow:feat/mps-supportCopy head branch name to clipboard