Skip to main content
`Ctrl`+`K`

# Array API Standard Support: `sparse`

## Support on CPU

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

blank = not currently documented

function/class | torch | jax | dask
--- | --- | --- | ---
block_array |  |  |
block_diag |  |  |
bmat |  |  |
bsr_array |  |  |
bsr_matrix |  |  |
coo_array |  |  |
coo_matrix |  |  |
csc_array |  |  |
csc_matrix |  |  |
csr_array |  |  |
csr_matrix |  |  |
dia_array |  |  |
dia_matrix |  |  |
diags |  |  |
diags_array |  |  |
dok_array |  |  |
dok_matrix |  |  |
expand_dims |  |  |
eye |  |  |
eye_array |  |  |
find |  |  |
get_index_dtype |  |  |
hstack |  |  |
identity |  |  |
issparse |  |  |
isspmatrix |  |  |
isspmatrix_bsr |  |  |
isspmatrix_coo |  |  |
isspmatrix_csc |  |  |
isspmatrix_csr |  |  |
isspmatrix_dia |  |  |
isspmatrix_dok |  |  |
isspmatrix_lil |  |  |
kron |  |  |
kronsum |  |  |
lil_array |  |  |
lil_matrix |  |  |
load_npz |  |  |
matrix_transpose |  |  |
permute_dims |  |  |
rand |  |  |
random |  |  |
random_array |  |  |
safely_cast_index_arrays |  |  |
save_npz |  |  |
sparray |  |  |
spdiags |  |  |
spmatrix |  |  |
swapaxes |  |  |
tril |  |  |
triu |  |  |
vstack |  |  |

## Support on GPU

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

blank = not currently documented

function/class | cupy | torch | jax
--- | --- | --- | ---
block_array |  |  |
block_diag |  |  |
bmat |  |  |
bsr_array |  |  |
bsr_matrix |  |  |
coo_array |  |  |
coo_matrix |  |  |
csc_array |  |  |
csc_matrix |  |  |
csr_array |  |  |
csr_matrix |  |  |
dia_array |  |  |
dia_matrix |  |  |
diags |  |  |
diags_array |  |  |
dok_array |  |  |
dok_matrix |  |  |
expand_dims |  |  |
eye |  |  |
eye_array |  |  |
find |  |  |
get_index_dtype |  |  |
hstack |  |  |
identity |  |  |
issparse |  |  |
isspmatrix |  |  |
isspmatrix_bsr |  |  |
isspmatrix_coo |  |  |
isspmatrix_csc |  |  |
isspmatrix_csr |  |  |
isspmatrix_dia |  |  |
isspmatrix_dok |  |  |
isspmatrix_lil |  |  |
kron |  |  |
kronsum |  |  |
lil_array |  |  |
lil_matrix |  |  |
load_npz |  |  |
matrix_transpose |  |  |
permute_dims |  |  |
rand |  |  |
random |  |  |
random_array |  |  |
safely_cast_index_arrays |  |  |
save_npz |  |  |
sparray |  |  |
spdiags |  |  |
spmatrix |  |  |
swapaxes |  |  |
tril |  |  |
triu |  |  |
vstack |  |  |

## Support with JIT

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

blank = not currently documented

function/class | jax
--- | ---
block_array |
block_diag |
bmat |
bsr_array |
bsr_matrix |
coo_array |
coo_matrix |
csc_array |
csc_matrix |
csr_array |
csr_matrix |
dia_array |
dia_matrix |
diags |
diags_array |
dok_array |
dok_matrix |
expand_dims |
eye |
eye_array |
find |
get_index_dtype |
hstack |
identity |
issparse |
isspmatrix |
isspmatrix_bsr |
isspmatrix_coo |
isspmatrix_csc |
isspmatrix_csr |
isspmatrix_dia |
isspmatrix_dok |
isspmatrix_lil |
kron |
kronsum |
lil_array |
lil_matrix |
load_npz |
matrix_transpose |
permute_dims |
rand |
random |
random_array |
safely_cast_index_arrays |
save_npz |
sparray |
spdiags |
spmatrix |
swapaxes |
tril |
triu |
vstack |