more_itertools
tqdm
torch>=1.11
torch_max_mem

[docs]
sphinx
sphinx-rtd-theme
sphinx-autodoc-typehints
sphinx_automodapi

[tests]
pytest
coverage
