pytorch-fast-transformers>=0.3.0
torch>=1.6
einops>=0.3
