
[cupy]
cupy

[dask]
dask

[jax]
jax

[numpy]
numpy

[pytorch]
pytorch

[sparse]
sparse>=0.15.1
