numpy>=1.15
jax
tqdm
brainstate>=0.5.1
brainunit>=0.2.0
brainevent>=0.0.7
braintools>=0.3.0
brainpy_state>=0.0.3

[cpu]
jax[cpu]
brainevent[cpu]

[cuda12]
jax[cuda12]
brainevent[cuda12]

[cuda13]
jax[cuda13]
brainevent[cuda13]

[tpu]
jax[tpu]
brainevent[tpu]
