numpy<=1.26.4
pandas
scikit-learn
torch
pytorch-lightning>1.6
lightning-bolts>0.4.0
