diff --git a/setup.py b/setup.py
index f13b500262..e4dc861ae6 100644
--- a/setup.py
+++ b/setup.py
@@ -123,6 +123,11 @@
     'grouped-gemm==0.1.4',
 ]
 
+extra_deps['databricks-serverless'] = {
+    dep for key, deps in extra_deps.items() for dep in deps
+    if 'gpu' not in key and 'megablocks' not in key and
+    'databricks-connect' not in dep
+}
 extra_deps['all-cpu'] = {
     dep for key, deps in extra_deps.items() for dep in deps
     if 'gpu' not in key and 'megablocks' not in key