New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Incompatibility with jax 0.4.27 #57

Open

tjltjl opened this issue May 7, 2024 · 1 comment

tjltjl commented May 7, 2024

Running the basic example produces:

when using the dependencies

Switching to 0.4.26 on both jax and jaxlib removes the error

Owner

patrick-kidger commented May 7, 2024

Thanks for the report! Looks like an upstream JAX bug. I've opened jax-ml/jax#21116.

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