patrick-kidger / optimistix

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

Fix some complex-related issues #53

Closed Randl closed 6 months ago

Randl commented 6 months ago

Related to https://github.com/patrick-kidger/diffrax/issues/388

patrick-kidger commented 6 months ago

This LGTM! If you can fix up the pre-commit failures then I'm happy to merge this as-is.

Randl commented 6 months ago

Should be fixed now

patrick-kidger commented 6 months ago

Great stuff! Merged.