Implementation of LSMR for iterative least squares.

lineax/__init__.py

@@ -58,6 +58,7 @@
     Cholesky as Cholesky,
     Diagonal as Diagonal,
     GMRES as GMRES,
+    LSMR as LSMR,
     LU as LU,
     NormalCG as NormalCG,
     QR as QR,

lineax/_solver/__init__.py

@@ -17,6 +17,7 @@
 from .cholesky import Cholesky as Cholesky
 from .diagonal import Diagonal as Diagonal
 from .gmres import GMRES as GMRES
+from .lsmr import LSMR as LSMR
 from .lu import LU as LU
 from .qr import QR as QR
 from .svd import SVD as SVD