diff --git a/pyproject.toml b/pyproject.toml
index 80af5e3..b21d605 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -51,6 +51,7 @@ pipelines = [
     "peft>=0.8.2",
     "trl>=0.7.10",
     "datasets",
+    "tensorboardX",
 ]
 # NOTE: When updating dependencies, in particular cuda/azure ml, make sure to update the azureml/conda.yaml too
 azure = ["azureml-core", "azureml-mlflow"]