matplotlib==3.5.1
numpy
pandas==1.4.2
scikit-learn==1.0.2
scipy==1.7.3
torch==1.11.0
tqdm==4.62.3
einops==0.4.1