Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
Here you can share your experience with the project you are suggesting or its comparison with optimistix. Optional.
A valid email to send you a verification link when necessary or log in.