torch
torchvision
einops
pytorch-lightning>=1.8
jsonargparse[signatures]>=4.12.0
torchmetrics[image]
wandb

[dev]
black
flake8
sphinx-autobuild
bumpver

[docs]
sphinx
myst_parser
furo
sphinxcontrib-katex

[tests]
pytest
