diff --git a/pyproject.toml b/pyproject.toml
index 0c5af635..af214053 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -22,7 +22,7 @@ packages = ["tsfm_public", "tsfm_public.toolkit", "tsfmhfdemos"]
 
 [project.optional-dependencies]
 
-notebooks = ["jupyter", "matplotlib", "datasets", "ipywidgets", "plotly", "kaleido"]
+notebooks = ["jupyter", "matplotlib", "datasets", "ipywidgets", "plotly", "kaleido", "tensorboard"]
 testing = ["pytest", "tsfm_public[notebooks]"]
 dev = ["pre-commit", "tsfm_public[testing]", "ruff==0.1.5"]
 evaluation = [