diff --git a/requirements.txt b/requirements.txt
index 8ee5920..195fb1d 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -5,3 +5,5 @@ tqdm
 more-itertools
 tiktoken
 triton>=2.0.0;platform_machine=="x86_64" and sys_platform=="linux" or sys_platform=="linux2"
+scipy
+pytest