iskra.sparse_linalg#

CUDSSSolver(mat[, analyze_only])

CholmodSolver(mat[, analyze_only, mode, ...])

Eigsh(*args, **kwargs)

LinearFactorAndSolve(*args, **kwargs)

Differentiable linear system.

build_diagonal_preconditioner(f, shape, ...)

build_sampled_diagonal_preconditioner(f, ...)

cg_solve(f, b, init, maxiter, tol)

Matrix-free Conjugate Gradient solver for (I - J^T) u = b.

default_solver(mat[, analyze_only, is_psd])

eigsh(A[, M, k, sigma, maxiter, tol, ...])

estimate_spectral_radius(f, init, maxiter)

Estimate spectral radius (max |evals|) of matrix J^T (given as callable f).

fixed_point_solver(f, z_init, maxiter[, tol])

gmres_solve(f, b, init, max_iter, abs_tol, ...)

Matrix-free GMRES solver for J^T u = b.

linear_solve(mat, b[, solver_fn])

make_power_iteration_vjp(a, evecs, m, sigma)

min_quadratic_energy(mat, rhs, known_idx, ...)

power_iteration(a, x, m, sigma[, solver])

power_iterations(A[, M, k, sigma, maxiter, tol])

quad_energy_mat(mat, unknown_idx)

quad_energy_rhs(mat, rhs, known_idx, ...[, ...])