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