jax>=0.9.0
numpy
scipy
matplotlib
saiunit>=0.5.2
brainstate>=0.5.0
braintools>=0.3.0

[cpu]
jax[cpu]>=0.9.0

[cuda12]
jax[cuda12]>=0.9.0

[cuda13]
jax[cuda13]>=0.9.0

[testing]
pytest
pytest-cov

[tpu]
jax[tpu]>=0.9.0
