numpy>=1.15
jax
tqdm
brainstate>=0.2.7
brainunit>=0.2.0
brainevent>=0.0.7
braintools>=0.0.9
brainpy_state>=0.0.3

[cpu]
jax[cpu]

[cuda12]
jax[cuda12]

[cuda13]
jax[cuda13]

[tpu]
jax[tpu]
