Unexpected behaviour with JAX version

patrick-kidger / optimistix

Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/

Apache License 2.0

334 stars 14 forks source link

Closed itk22 closed 6 months ago

itk22 commented 6 months ago

Hi Patrick, I have run into an peculiar behaviour when trying to install optimistix with jax 0.4.25:

It seems to upgrade to 0.4.28 despite 0.4.25 already satisfying the requirement. Is this the expected behaviour?

itk22 commented 6 months ago

I think I might have found the reason - the newest version of lineax require jax 0.4.26 which is quite new. Is there any particular reason for this?

patrick-kidger commented 6 months ago

Yup. JAX version 0.4.26 fixed a correctness issue when using complex numbers, that we were hitting in Lineax.

If you really need to use an earlier version, and you're not using complex numbers, then you can probably get away with it though :)

itk22 commented 6 months ago

Ahh, I see. Thanks for the super fast response!