numpy>=1.15
jax
tqdm
brainstate>=0.2.7
brainunit
brainevent>=0.0.4
braintools>=0.0.9
brainpy_state>=0.0.2

[cpu]
jax[cpu]

[cuda12]
jax[cuda12]

[cuda13]
jax[cuda13]

[tpu]
jax[tpu]
