numpy>=2.2.6
einops>=0.8.2
torch>=2.0.0
