torch>=1.13
einops
numpy
