From adb975c1993a9e155221f2cbf77d3da1b7e1cd63 Mon Sep 17 00:00:00 2001 From: Kashif Rasul Date: Tue, 10 May 2022 10:58:03 +0200 Subject: [PATCH] added s4 requirements --- requirements.txt | 4 ++++ 1 file changed, 4 insertions(+) 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