Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
Why do you think that https://github.com/keras-team/keras is a good alternative to optimistix
Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
Why do you think that https://github.com/keras-team/keras is a good alternative to optimistix