dataclasses-json
numpy
scipy>=1.7.0
click
click_default_group
scikit-learn
torch>=2.0
tqdm
requests
optuna>=2.0.0
pandas>=1.0.0
tabulate
more_click
more_itertools
pystow>=0.4.3
docdata
class_resolver>0.4.2
pyyaml
torch_max_mem>=0.1.1
torch-ppr>=0.0.7
typing_extensions

[biomedicine]
bioregistry
pyobo>=0.8.7

[docs]
sphinx<7
sphinx-rtd-theme
sphinx-click
sphinx-autodoc-typehints<=1.13.1
sphinx_automodapi
texext

[lightning]
pytorch_lightning>=1.7.2

[mlflow]
mlflow>=1.8.0

[neptune]
neptune-client

[ogb]
ogb>=1.3.4

[opt_einsum]
opt_einsum

[plotting]
matplotlib
seaborn

[templating]
jinja2

[tensorboard]
tensorboard

[tests]
unittest-templates>=0.0.5
coverage
pytest

[transformers]
transformers

[wandb]
wandb

[wordcloud]
wordcloud
IPython
