torch<1.8.1
pytorch-lightning>=1.4.5
sparseml
