torch == 2.3.1
numpy == 1.23.5
pandas == 1.5.3
einops==0.4.0
tqdm==4.64.1