search

patrick-kidger / optimistix

Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
Apache License 2.0
312 stars 15 forks source link

Added `optimistix.compat.minimize` as a replacement for `jax.scipy.optimize.minimize` #14

Closed patrick-kidger closed 11 months ago

patrick-kidger commented 11 months ago

This API is being removed from JAX in favour of the domain-specific libraries Optimistix and JAXopt.

As such this PR adds in a drop-in compatibility layer to Optimistix.