numpy==1.26.4
torch==2.4.0
einops==0.8.0