torch~=2.3.1
einops~=0.8.0
numpy
