diff --git a/requirements.txt b/requirements.txt
index 1e7475d71..8e781c022 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -26,15 +26,12 @@ dm-tree==0.1.8
 dp-accounting==0.4.3
 immutabledict~=2.2
 matplotlib~=3.3
-mpmath~=1.2
 numpy~=1.21
 packaging~=22.0
 pandas~=1.4
-scikit-learn>=1.0,==1.*
 scipy~=1.9
 statsmodels~=0.13
 tensorflow-datasets~=4.5
 tensorflow-estimator~=2.4
 tensorflow-probability~=0.20.0
 tensorflow~=2.4
-tf-models-official~=2.13
diff --git a/setup.py b/setup.py
index 4928c8cca..a05c127d7 100644
--- a/setup.py
+++ b/setup.py
@@ -31,20 +31,12 @@
         'attrs>=21.4',
         'dm-tree==0.1.8',
         'dp-accounting==0.4.3',
-        'immutabledict~=2.2',
-        'matplotlib~=3.3',
         'numpy~=1.21',
         'packaging~=22.0',
-        'pandas~=1.4',
-        'parameterized~=0.8',
-        'scikit-learn>=1.0,==1.*',
         'scipy~=1.9',
-        'statsmodels~=0.13',
-        'tensorflow-datasets~=4.5',
         'tensorflow-estimator~=2.4',
         'tensorflow-probability~=0.20.0',
         'tensorflow~=2.4',
-        'tf-models-official~=2.13',
     ],
     packages=find_packages(),
 )