numpy
torch
torchvision
tqdm
omegaconf
wandb
hydra-core
hydra-submitit-launcher
pandas
jsonlines
torchmetrics
scipy
datasets

[dev]
pytest
coverage
numpydoc
pytest-cov
codecov
pre-commit
ruff

[doc]
sphinx
sphinx-gallery
sphinx-book-theme
sphinxcontrib-bibtex
matplotlib
numpydoc
memory_profiler
myst-parser
