diff --git a/requirements.txt b/requirements.txt
index 3747eecff..deb7132f0 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -21,3 +21,4 @@ ml_dtypes
 psutil
 scipy
 tornado
+torch