einops>=0.3
local-attention>=1.1.1
torch>=1.6
axial-positional-embedding>=0.1.0
