torch>=1.6.0
einops
x-transformers
