jax
numpy>=1.15
brainstate>=0.1.0
brainunit>=0.0.8
typing_extensions
optax
matplotlib
msgpack
scipy

[cogtask]
brainstate>=0.1.0
brainunit>=0.0.8

[cpu]
jax[cpu]

[cuda12]
jax[cuda12]

[cuda13]
jax[cuda13]

[doc]
pandoc
Jinja2
sphinx>=9.0.4
myst-nb
sphinx_thebe
sphinx-autodoc-typehints
sphinx-book-theme>=1.2.0
sphinx-copybutton>=0.5.2
sphinx-remove-toctrees
jupyter-sphinx>=0.5.3
sphinx-design
sphinx_math_dollar
brainx-sphinx-header>=0.4.0

[testing]
pytest
absl-py
nevergrad

[tpu]
jax[tpu]
