torch>=2.0
pytorch-lightning<2.1,>=2.0
torchmetrics>=0.11.4
matplotlib>=3.3
pyyaml>=5.4
