diff --git a/requirements.txt b/requirements.txt
index 27c0b7c..b66eb78 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,10 @@
 einops
 jaxtyping
 schedulefree
-liger-kernel
\ No newline at end of file
+liger-kernel
+
+# visualization
+tensorboard
+flask
+pandas
+plotly