pytorch-lightning>=1.7.0
lightning-utilities>0.4.0, <0.5.0  # this is needed for PL 1.7
