Array API Standard Support: sparse.linalg#

Support on CPU#

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

blank = not currently documented

function/class

torch

jax

dask

LaplacianNd

LinearOperator

✔️

✔️

✔️

SuperLU

aslinearoperator

✔️

✔️

✔️

bicg

bicgstab

cg

cgs

eigs

eigsh

expm

expm_multiply

factorized

funm_multiply_krylov

gcrotmk

gmres

inv

is_sptriangular

lgmres

lobpcg

lsmr

lsqr

matrix_power

minres

norm

onenormest

qmr

spbandwidth

spilu

splu

spsolve

spsolve_triangular

svds

tfqmr

use_solver

Support on GPU#

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

blank = not currently documented

function/class

cupy

torch

jax

LaplacianNd

LinearOperator

✔️

✔️

✔️

SuperLU

aslinearoperator

✔️

✔️

✔️

bicg

bicgstab

cg

cgs

eigs

eigsh

expm

expm_multiply

factorized

funm_multiply_krylov

gcrotmk

gmres

inv

is_sptriangular

lgmres

lobpcg

lsmr

lsqr

matrix_power

minres

norm

onenormest

qmr

spbandwidth

spilu

splu

spsolve

spsolve_triangular

svds

tfqmr

use_solver

Support with JIT#

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

blank = not currently documented