diff --git a/docker/dockerfile.merlin b/docker/dockerfile.merlin
index 577a7ffed..b844e64b6 100644
--- a/docker/dockerfile.merlin
+++ b/docker/dockerfile.merlin
@@ -95,7 +95,7 @@ RUN pip install "cmake<3.25.0" ninja scikit-build
 RUN pip install pandas==1.3.5
 RUN pip install cupy-cuda117 nvidia-pyindex pybind11 pytest protobuf transformers==4.12 tensorflow-metadata
 RUN pip install betterproto cachetools graphviz nvtx scipy sklearn
-RUN pip install tritonclient[all] grpcio-channelz fiddle wandb npy-append-array
+RUN pip install tritonclient[all] grpcio-channelz fiddle wandb npy-append-array optuna plotly
 RUN pip install git+https://github.com/rapidsai/asvdb.git@main
 # xgboost and treelite are pinned from tritonserver support
 RUN pip install xgboost==1.6.2 lightgbm treelite==2.3.0 treelite_runtime==2.3.0