torch~=1.11.0
setuptools~=59.2.0
einops~=0.4.1
