einops==0.6.0 matplotlib==3.6.3 numpy==1.24.1 omegaconf==2.3.0 scikit_learn==1.2.1 scipy==1.10.0 torch==1.13.1 tqdm==4.64.1