torch
axial-positional-embedding
product-key-memory>=0.1.5
