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

[dev]
black
flake8
sphinx-autobuild

[docs]
sphinx
myst_parser
furo
sphinxcontrib-katex

[tests]
pytest
