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 | torch | jax | dask
--- | --- | --- | ---
block_array |  |  |
block_diag |  |  |
bmat |  |  |
diags |  |  |
diags_array |  |  |
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 |  |  |
load_npz |  |  |
permute_dims |  |  |
rand |  |  |
random |  |  |
random_array |  |  |
safely_cast_index_arrays |  |  |
save_npz |  |  |
spdiags |  |  |
swapaxes |  |  |
tril |  |  |
triu |  |  |
vstack |  |  |

## Support on GPU

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

blank = not currently documented

function | cupy | torch | jax
--- | --- | --- | ---
block_array |  |  |
block_diag |  |  |
bmat |  |  |
diags |  |  |
diags_array |  |  |
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 |  |  |
load_npz |  |  |
permute_dims |  |  |
rand |  |  |
random |  |  |
random_array |  |  |
safely_cast_index_arrays |  |  |
save_npz |  |  |
spdiags |  |  |
swapaxes |  |  |
tril |  |  |
triu |  |  |
vstack |  |  |

## Support with JIT

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

blank = not currently documented

function | jax
--- | ---
block_array |
block_diag |
bmat |
diags |
diags_array |
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 |
load_npz |
permute_dims |
rand |
random |
random_array |
safely_cast_index_arrays |
save_npz |
spdiags |
swapaxes |
tril |
triu |
vstack |