numpy
pandas
seaborn
xarray>=0.18.0
scikit-learn
tqdm
jaxlib>=0.1.67
jax
dm-haiku>=0.0.4

[complete]
torch
torchvision
tensorflow-datasets
tensorflow>=2.4.0
optax
plotnine
pyarrow
pooch
mypy
ipykernel
ipywidgets
nbconvert
pytest
pytest-cov
pytest-xdist
pytest-benchmark
flaky
coverage
flake8
autopep8
autoflake
black
isort>=5.3.0
jupytext>=1.11.1
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.11.1
papermill
line_profiler

[docs]
torch
torchvision
tensorflow-datasets
tensorflow>=2.4.0
optax
plotnine
pyarrow
pooch
mypy
ipykernel
ipywidgets
nbconvert
pytest
pytest-cov
pytest-xdist
pytest-benchmark
flaky
coverage
flake8
autopep8
autoflake
black
isort>=5.3.0
jupytext>=1.11.1
papermill
line_profiler
sphinx
sphinxcontrib-napoleon
sphinx_rtd_theme
sphinx-autosummary-accessors
ipython
ipykernel
jupyter-client
nbsphinx
scanpydoc

[optional]
torch
torchvision
tensorflow-datasets
tensorflow>=2.4.0
optax
plotnine
pyarrow
pooch
