Skip to content

Lineax v0.0.2

Compare
Choose a tag to compare
@github-actions github-actions released this 30 Sep 02:16
· 67 commits to main since this release

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 now IdentityLinearOperator(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