diff --git a/requirements.txt b/requirements.txt index 3c93a6b..68c28f0 100644 --- a/requirements.txt +++ b/requirements.txt @@ -6,3 +6,7 @@ xformers https://github.com/ml-jku/hopfield-layers etsformer-pytorch reformer_pytorch +einops +opt_einsum +pykeops +scipy