jax>=0.4.28
numpy>1.20
scipy>=1.10
typing_extensions>=4.6
pynapple>=0.10.1
scikit-learn
optimistix>=0.0.11
optax>=0.2.5

[dev]
black
isort
pip-tools
ruff
pytest
pytest-xdist
flake8
coverage
pytest-cov
statsmodels
scikit-learn
matplotlib>=3.7
pooch
dandi
seaborn
myst-nb
hmmlearn
tox-uv

[docs]
numpydoc
sphinx
pydata-sphinx-theme
sphinx-autodoc-typehints
sphinx-copybutton
sphinx-design
sphinx-issues
sphinxcontrib-apidoc
sphinxcontrib-bibtex
sphinx-togglebutton
sphinx_code_tabs
sphinxemoji
myst-parser
myst-nb
dandi
sphinx-autobuild
sphinx-contributors
scikit-learn
matplotlib>=3.7
seaborn
pooch
ipywidgets
ipykernel
patsy
tqdm

[examples]
scikit-learn
dandi
matplotlib>=3.7
seaborn
pooch
fsspec
patsy
tqdm

[jaxopt]
jaxopt>=0.6
