torch<1.14,>=1.13.0
numpy<1.26,>=1.21.0
lightning<2.1,>=2.0
scikit-learn<2.0,>=1.2.0
pandas<1.6.0,>=1.4.0
einops<1.0.0,>=0.6.0
rdata<1.0.0,>=0.9.0
types-requests
optuna<3.4.0,>=3.1.0
