numpy<1.22,>=1.18
pandas
seaborn
xarray>=0.18.0
scikit-learn
tqdm
jaxlib
jax
dm-haiku>=0.0.4

[complete]
optax
plotnine
pyarrow
pooch
graphviz
numba
mypy
ipykernel
ipywidgets
nbconvert
pytest
pytest-cov
pytest-xdist
pytest-benchmark
flaky
coverage
flake8
autopep8
autoflake
black
isort>=5.3.0
jupytext<=1.13.3
papermill
line_profiler

[dev]
mypy
ipykernel
ipywidgets
nbconvert
pytest
pytest-cov
pytest-xdist
pytest-benchmark
flaky
coverage
flake8
autopep8
autoflake
black
isort>=5.3.0
jupytext<=1.13.3
papermill
line_profiler

[docs]
optax
plotnine
pyarrow
pooch
graphviz
numba
mypy
ipykernel
ipywidgets
nbconvert
pytest
pytest-cov
pytest-xdist
pytest-benchmark
flaky
coverage
flake8
autopep8
autoflake
black
isort>=5.3.0
jupytext<=1.13.3
papermill
line_profiler
sphinx
sphinxcontrib-napoleon
sphinx_rtd_theme
sphinx-autodoc-typehints
sphinx-autosummary-accessors
ipython
jupyter-client
jupyter-sphinx
myst-nb
nbsphinx
scanpydoc
matplotlib
sklearn

[optional]
optax
plotnine
pyarrow
pooch
graphviz
numba
