diff --git a/setup.py b/setup.py
index 0b7e512955..b3b6e2e063 100644
--- a/setup.py
+++ b/setup.py
@@ -38,7 +38,9 @@ from setuptools import find_packages, setup
 
 
 extras = {
-    'serving': ['uvicorn', 'fastapi']
+    'serving': ['uvicorn', 'fastapi'],
+    'serving-tf': ['uvicorn', 'fastapi', 'tensorflow'],
+    'serving-torch': ['uvicorn', 'fastapi', 'torch']
 }
 extras['all'] = [package for package in extras.values()]