Skip to content

Can't instantiate jax numpy zeros if 64-bit mode is enabled using GPU #18392

Answered by jakevdp
nelsen4 asked this question in General

You must be logged in to vote

Oh, I see you mention jax-metal above.

jax-metal is highly experimental, and does not support the full JAX API. You can see the current list of reported metal-related issues here: https://github.com/google/jax/issues?q=is%3Aopen+is%3Aissue+label%3A%22Apple+GPU+%28Metal%29+plugin%22

It sounds like you're running into the issue previously reported in #16435

Note that jax-metal is closed source and the JAX core team has no access to the code, so there's not much we can do about issues like this.

Replies: 1 comment 4 replies

You must be logged in to vote
4 replies
@nelsen4

@jakevdp

@jakevdp

Answer selected by nelsen4
@nelsen4

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment