Fix JAX deprecation warning

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 patrick-kidger closed 5 months ago