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