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

[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_dev]
pytest
absl-py>=0.7.0
cvxpy>=1.1.11
optax>=0.0.9
pytest-xdist
scikit-learn>=0.24.1
cvxopt
