einops>=0.8
torch>=2.3
