[tx] Enable MPS backend for Apple Silicon on Jax backend#1332
Open
pcmoritz wants to merge 1 commit intoNovaSky-AI:mainfrom
Open
[tx] Enable MPS backend for Apple Silicon on Jax backend#1332pcmoritz wants to merge 1 commit intoNovaSky-AI:mainfrom
pcmoritz wants to merge 1 commit intoNovaSky-AI:mainfrom