einops>=0.3
local-attention>=1.1.1
pytorch-fast-transformers>=0.3.0
torch>=1.6
