diff --git a/requirements.txt b/requirements.txt
index 762db4f34..e4e5ec642 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -28,3 +28,4 @@ kornia
 lark
 inflection
 GitPython
+torchsde