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