torch>=2.0
torchvision
numpy
setuptools>=65.5.0
pytorch_lightning
av
jsonargparse[signatures]
einops

[test]
pytest
coverage
flake8
black
isort
pytest-cov
mypy
gitchangelog
mkdocs
lark
types-requests
requests-mock
