numpy
typing_extensions
array_api_compat>=1.9
opt_einsum

[all]
saiunit[cupy,dask,jax,ndonnx,torch]

[cpu]
jax[cpu]>=0.9.0

[cuda12]
jax[cuda12]>=0.9.0

[cuda13]
jax[cuda13]>=0.9.0

[cupy]
cupy-cuda12x>=13.0

[dask]
dask[array]>=2024.1

[jax]
jax>=0.9.0

[ndonnx]
ndonnx>=0.9

[testing]
pytest
brainstate

[torch]
torch>=2.0

[tpu]
jax[tpu]>=0.9.0
