diff --git a/megatron/requirements.txt b/megatron/requirements.txt
index d5d8ff4..18fda94 100644
--- a/megatron/requirements.txt
+++ b/megatron/requirements.txt
@@ -1,5 +1,9 @@
+numpy==1.23.4
 pybind11==2.13.6
 pyyaml==6.0.2
 regex==2024.9.11
+tensorboard==2.18.0
+tensorboard-data-server==0.7.2
 transformers==4.45.2
-triton==3.0.0
\ No newline at end of file
+triton==3.0.0
+wandb==0.18.3