numpy>=1.15
brainstate>=0.1.0
brainunit>=0.0.8
typing_extensions
optax

[cpu]
jax[cpu]
brainstate
brainunit
matplotlib
scipy
msgpack
optax
typing_extensions

[cuda12]
jax[cuda12]
brainstate
brainunit
matplotlib
scipy
msgpack
optax
typing_extensions

[cuda13]
jax[cuda13]
brainstate
brainunit
matplotlib
scipy
msgpack
optax
typing_extensions

[testing]
jax
brainstate
brainunit
matplotlib
nevergrad
scipy
msgpack
optax
pytest
absl-py
typing_extensions

[tpu]
jax[tpu]
brainstate
brainunit
matplotlib
scipy
msgpack
optax
typing_extensions
