search

patrick-kidger / lineax

Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax
Apache License 2.0
365 stars 24 forks source link

Change type annotation to allow complex types #59

Closed Randl closed 1 year ago

Randl commented 1 year ago

Replace Float with Inexact that allows complex datatypes. See https://github.com/google/lineax/issues/57

patrick-kidger commented 1 year ago

LGTM!