diff --git a/clu/metric_writers/tf/summary_writer.py b/clu/metric_writers/tf/summary_writer.py
index 5711e70..081e8e9 100644
--- a/clu/metric_writers/tf/summary_writer.py
+++ b/clu/metric_writers/tf/summary_writer.py
@@ -25,9 +25,13 @@
 
 from clu.internal import utils
 from clu.metric_writers import interface
+from etils import epy
 import tensorflow as tf
 
-from tensorboard.plugins.hparams import api as hparams_api
+with epy.lazy_imports():
+  # pylint: disable=g-import-not-at-top
+  from tensorboard.plugins.hparams import api as hparams_api
+  # pylint: enable=g-import-not-at-top
 
 
 Array = interface.Array