torch>=1.6
data-science-types>=0.2
einops>=0.6.0
xformers>=0.0.13
