axial-positional-embedding
einops
linformer>=0.1.0
local-attention
product-key-memory>=0.1.5
torch
