torch_spsolve. A package that provides differentiable sparse solves in pytorch, both in the sparse operator and right hand side (rhs).