diff --git a/setup.py b/setup.py
index f528838d35..81178686d2 100644
--- a/setup.py
+++ b/setup.py
@@ -83,6 +83,10 @@
     'hf_transfer==0.1.3',
 ]
 
+extra_deps['databricks'] = [
+    'mosaicml[databricks]',
+]
+
 extra_deps['tensorboard'] = [
     'mosaicml[tensorboard]>=0.16.1,<0.17',
 ]