POT
PyYAML
chex
jax
jaxopt
numpy
optax
pandas
scipy
toolz
tqdm

[all]
h5py
gpu-tracker
scikit-learn
Pillow
scikit-image
matplotlib
plotly
seaborn
dataframe-image
ipython
dash
opencv-python

[color-transfer]
dash
opencv-python
Pillow
scikit-image
plotly
matplotlib

[cuda12]
jax[cuda12]

[dev]
pytest>=7.0
pytest-cov
coverage
black>=23.3.0
ruff>=0.4.1
pyright

[docs]
mkdocs>=1.5
mkdocs-material>=9.5
mkdocstrings[python]>=0.24

[gurobi]
gurobipy

[image-analysis]
Pillow
scikit-image
matplotlib
seaborn
dataframe-image
ipython

[lint]
black>=23.3.0
ruff>=0.4.1

[mnist]
scikit-learn

[profiling]
gpu-tracker

[storage]
h5py

[testing]
pytest>=7.0

[viz]
matplotlib
plotly
seaborn
dataframe-image
ipython
