diff --git a/requirements.txt b/requirements.txt
index b51d6ac..9a47472 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -35,5 +35,5 @@ scipy~=1.7
 statsmodels~=0.13
 tensorflow-datasets~=4.5
 tensorflow-estimator~=2.4
-tensorflow-probability==0.15.0
+tensorflow-probability~=0.20.0
 tensorflow~=2.4
diff --git a/setup.py b/setup.py
index 9befb55..58e3330 100644
--- a/setup.py
+++ b/setup.py
@@ -42,7 +42,7 @@ setup(
         'statsmodels~=0.13',
         'tensorflow-datasets~=4.5',
         'tensorflow-estimator~=2.4',
-        'tensorflow-probability==0.15.0',
+        'tensorflow-probability~=0.20.0',
         'tensorflow~=2.4',
     ],
     packages=find_packages(),