torch>=2.0.0
numpy>=1.24.3
torchopt>=0.7.1
lightning>=2.0.2
torchmetrics>=1.2.0
omegaconf>=2.3.0
