einops==0.8.1
matplotlib==3.10.3
ml-collections
numpy==2.2.6
pyyaml==6.0.2
scikit-learn==1.7.1
scipy==1.15.3
torch==2.1.0+cu121
tqdm==4.67.1
