numpy>=1.15
jax>=0.9.0
tqdm
brainstate>=0.5.1
brainunit>=0.2.0
brainevent>=0.2.1
braintools>=0.3.0

[cpu]
jax[cpu]>=0.9.0
brainevent[cpu]

[cuda12]
jax[cuda12]>=0.9.0
brainevent[cuda12]

[cuda13]
jax[cuda13]>=0.9.0
brainevent[cuda13]

[tpu]
jax[tpu]>=0.9.0
brainevent[tpu]
