diff --git a/requirements.txt b/requirements.txt
index 24be600..31929da 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -9,4 +9,5 @@ scipy~=1.2.1
 sklearn~=1.0.2
 tensorflow-datasets>=4.4.0
 tensorflow-estimator>=2.3.0
+tensorflow-probability>=0.13.0
 tensorflow>=1.14
diff --git a/setup.py b/setup.py
index f70ad25..393b85f 100644
--- a/setup.py
+++ b/setup.py
@@ -37,6 +37,7 @@ setup(
         'sklearn~=1.0.2',
         'tensorflow-datasets>=4.4.0',
         'tensorflow-estimator>=2.3.0',
+        'tensorflow-probability>=0.13.0',
     ],
     # Explicit dependence on TensorFlow is not supported.
     # See https://github.com/tensorflow/tensorflow/issues/7166