patrick-kidger / optimistix

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

Fix some complex-related issues #53

Closed Randl closed 7 months ago

Randl commented 8 months ago

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

patrick-kidger commented 7 months ago

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

Randl commented 7 months ago

Should be fixed now

patrick-kidger commented 7 months ago

Great stuff! Merged.