einops>=0.6.0
local-attention>=1.8.5
torch>=1.10
