numpy
typing_extensions
saiunit==0.2.2

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

[cpu]
jax[cpu]

[cuda12]
jax[cuda12]

[cuda13]
jax[cuda13]

[cupy]
cupy-cuda12x>=13.0

[dask]
dask[array]>=2024.1

[jax]
jax

[ndonnx]
ndonnx>=0.9

[testing]
pytest
brainstate

[torch]
torch>=2.0

[tpu]
jax[tpu]
