einops>=0.8
torch>=2.4
