diff --git a/pkgs/development/python-modules/ax-platform/default.nix b/pkgs/development/python-modules/ax-platform/default.nix
index aee136f5c4e8..6254cf13b416 100644
--- a/pkgs/development/python-modules/ax-platform/default.nix
+++ b/pkgs/development/python-modules/ax-platform/default.nix
@@ -28,6 +28,7 @@
   pytestCheckHook,
   sqlalchemy,
   tabulate,
+  tensorboard,
 }:
 
 buildPythonPackage (finalAttrs: {
@@ -72,6 +73,7 @@ buildPythonPackage (finalAttrs: {
     pytestCheckHook
     sqlalchemy
     tabulate
+    tensorboard
   ];
 
   disabledTestPaths = [