numpy>=2.0
brainunit>=0.0.8
jax>=0.8.0

[cpu]
jax[cpu]>=0.8.0
numba

[cuda12]
jax[cuda12]>=0.8.0

[cuda13]
jax[cuda13]>=0.8.0

[testing]
pytest
pytest-cov

[tpu]
jax[tpu]>=0.8.0
