torch>=1.6
torchmetrics>=0.4.1
pytorch-lightning>=1.3.0
dataclasses ; python_version <= "3.6"
packaging
