brainstate>=0.2.2
brainunit
brainstate
brainpy>=2.7.1
braintools

[cpu]
jax[cpu]
brainunit
brainstate
brainpy
braintools

[cuda12]
jax[cuda12]
brainunit
brainstate
brainpy
braintools

[cuda13]
jax[cuda13]
brainunit
brainstate
brainpy
braintools

[testing]
pytest
jax[cpu]
brainunit
brainstate
brainpy
braintools

[tpu]
jax[tpu]
brainunit
brainstate
brainpy
braintools
