Apple M1 GPU support on JAX? #9847
-
|
Just wondering if JAX might create support for Apple M1 GPU cores? So far Jax has worked fine with M1 CPU cores on MacBookPro M1 max. thanks. |
Beta Was this translation helpful? Give feedback.
Replies: 3 comments
-
|
I think the discussion about this is at #8074. Does that sound right to you? |
Beta Was this translation helpful? Give feedback.
-
|
Thank you much. I followed the discussion thread and it looks promising, especially with the shark stuff on PyTorch. |
Beta Was this translation helpful? Give feedback.
-
|
Looks like it is now available. |
Beta Was this translation helpful? Give feedback.
Looks like it is now available.
https://developer.apple.com/metal/jax/