diff --git a/setup.py b/setup.py
index b79e3ab..74a75fd 100644
--- a/setup.py
+++ b/setup.py
@@ -35,4 +35,7 @@
         "tqdm>=4.67",
         "websockets>=14.0",
     ],
+    extras_require={
+        "ml": ["dataproc-ml>=1.0.0-rc1,<=1.0"],
+    },
 )