torch>=2.7.0
einops
