torch>=1.0
torchmetrics>=0.10.0
pytorch-lightning
lightning-utilities
