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