einops==0.8.0
ml_collections==1.0.0
numpy==1.23.5
scikit-learn==1.2.2
scipy==1.15.1
pandas
matplotlib