beartype
einops>=0.6.1
memorizing-transformers-pytorch>=0.4.0
torch>=1.6
