Jvp and helper fixes - Githubissues

patrick-kidger / lineax

Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax

Apache License 2.0

365 stars 24 forks source link

Jvp and helper fixes #66

Closed Randl closed 11 months ago

Randl commented 11 months ago

Since https://github.com/google/lineax/pull/64 is blocked by a bug, I'm taking functional changes to a separate PR (but not enabling tests for complex variables yet). This includes fixing JVP for linear_solve, fixing Cholesky solver transposition, changing the randomness source in helper to be reproducible, and adding dtype parametrization to more tests.

patrick-kidger commented 11 months ago

Alright, this LGTM! Thanks for these fixes, we're now much closer to full complex support.