numpy>=1.15
jax
brainpy>=2.7.6
brainstate>=0.5.0
brainunit
brainevent>=0.0.4
braintools>=0.0.9

[cpu]
jax[cpu]

[cuda12]
jax[cuda12]

[cuda13]
jax[cuda13]

[tpu]
jax[tpu]
