lineax

lineax copied to clipboard

patrick-kidger

Block operators / solvers

1

Following the discussion in #80 I thought it would be good to lay out some ideas for a general block matrix/operator abstraction. Some useful types of block operators: - Block...

f0uriest

Added tests for #101. Requires https://github.com/patrick-kidger/equinox/pull/795 (so these tests will fail right now)

patrick-kidger

In issue #3 , a solver based on Woodbury matrix identity was given as a new solver feature for lineax. Here I have made a first attempt at implementing it....

aidancrilly

Bug in QR solver for Tikhnov regularised matrices

2

When solving against matrices of the form $A + \lambda I$ with $\lambda$ set to the max dtype value, QR solves return `nan` instead of zero. This does not occur...

packquickly

Creating JacobianLinearOperator fails with confusing error

2

I'm getting a confusing error when trying to create a `JacobianLinearOperator`. - `lx.JacobianLinearOperator(f, x0)` causes `TypeError: f() takes 1 positional argument but 2 were given`, - `lx.JacobianLinearOperator(f)` causes `TypeError: JacobianLinearOperator.__init__()...

johannahaffner

Failure to raise error for ill-posed system if using LU and 32-bit

2

When using `AutoLinearOperator` with `well_posed=True` and no tags specifying some structure, the LU solver is used. LU wraps the `jax.scipy` implementation and raises an error if the output has `NaN`...

johannahaffner

Differentiating w.r.t. initial guess throws an error

2

The gradient of the solution of a linear system solved iteratively w.r.t. to the initial guess should be zero. Instead, the following snippet ```python import lineax as lx from jax...

romanodev