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