numpy>=1.15
jax
tqdm
brainstate>=0.2.0
brainunit
brainevent>=0.0.4
braintools>=0.0.9

[cpu]
jax[cpu]
brainstate
brainunit
brainevent
braintools

[cuda12]
jax[cuda12]
brainstate
brainunit
brainevent
braintools

[cuda13]
jax[cuda13]
brainstate
brainunit
brainevent
braintools

[tpu]
jax[tpu]
brainstate
brainunit
brainevent
braintools
