Post marked as solved
Click to stop watching this thread.
You have stopped watching this post. Click to start watching again.
contentPostList.repliessolved.tooltip
Replied In
Metal JAX device appears as `cpu`
JAX v0.4.11 is required, so fixed by any of the following suggestions:
https://developer.apple.com/forums/thread/731465?answerId=756330022#756330022
https://developer.apple.com/forums/thread/731465?answerId=756319022#756319022
(JAX v04.12 doesn't work with Metal. v0.4.13 requires a newer version of jaxlib)