torch>=2.2.1
pytorch_optimizer>=3.0.0
