optimistix icon indicating copy to clipboard operation
optimistix copied to clipboard
verified publisher icon patrick-kidger

Incompatibility with jax 0.4.27

Open tjltjl opened this issue 1 year ago • 1 comments

Running the basic example produces:

image

when using the dependencies image

Switching to 0.4.26 on both jax and jaxlib removes the error

tjltjl avatar May 07 '24 20:05 tjltjl

Thanks for the report! Looks like an upstream JAX bug. I've opened https://github.com/google/jax/issues/21116.

patrick-kidger avatar May 07 '24 20:05 patrick-kidger