numpy
jax
brainstate>=0.2.9
brainunit
braintools>=0.3.0

[cpu]
jax

[cuda12]
jax[cuda12]

[cuda13]
jax[cuda13]

[testing]
pytest
jax

[tpu]
jax[tpu]

[viz]
matplotlib
