Sparse triangular solve without QR factorization

[hanbin973]

I have a sparse triangular system to solve. Is it possible to solve this system directly by back/forward substitution without QR factorization in jax?

[dfm]

I think that lax.linalg.triangular_solve might do the trick for you?

[hanbin973]

I checked the function already but it only takes dense Jax arrays. My sparse system is too large to fill in all the zeros.