Lineax v0.0.2
Minor release, just tweaking a few things.
- Fixed all the warnings that get printed out using Lineax v0.0.1 with Equinox v0.11.0. (Equinox is correctly shouting at us that we screwed up by defining both
__init__
and__post_init__
.) IdentityLinearOperator(structure)
is nowIdentityLinearOperator(input_structure, output_structure)
, to support differing input/output structures.- Linear operators now treat their tags (for symmetry/positive-definite-ness/etc.) as static fields. This means that e.g. operators can now be used as the carry in a
lax.scan
. lineax.SVD()
now no longer returns NaNs when given a matrix of all zeros. (This is also an improvement over core JAX, which does not handle this case.)- Some performance improvements to autodiff-of-
linear_solve
. - Fixed autodiff-of-
linear_solve
crashing in a handful of edge cases. (When JAX passed symbolic zeros for AD.) - Many documentation fixes.
New Contributors
Full Changelog: v0.0.1...v0.0.2