diff --git a/requirements.txt b/requirements.txt
index fa23f91..2786f83 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -18,4 +18,15 @@ python-dotenv
 ccxt
 yfinance
 flask_cors
-safetensors
\ No newline at end of file
+safetensors
+torch>=2.1
+lightning>=2.0
+gluonts~=0.14.3
+# numpy~=1.26.0
+einops==0.7.*
+jaxtyping~=0.2.24
+python-dotenv
+hydra-core==1.3
+huggingface_hub>=0.23.0
+safetensors
+jax[cpu]
\ No newline at end of file