dataclasses-json
numpy
scipy>=1.7.0
click
click_default_group
sklearn
tqdm
requests
optuna>=2.0.0
pandas>=1.0.0
tabulate
more_click
more_itertools
pystow>=0.4.3
docdata
class_resolver>=0.3.9
pyyaml
rexmex
torch_max_mem
torch-ppr>=0.0.4

[:platform_system != "Windows"]
torch>=1.10

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

[lightning]
pytorch_lightning

[mlflow]
mlflow>=1.8.0

[neptune]
neptune-client

[plotting]
matplotlib
seaborn

[templating]
jinja2

[tensorboard]
tensorboard

[tests]
unittest-templates>=0.0.5
coverage
pytest

[transformers]
transformers

[wandb]
wandb
