torch>=2.5
einops
numpy
