diff --git a/requirements.txt b/requirements.txt
index 47ea8f8..ea753f2 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,5 @@
 tensorflow==2.1.0
+tensorflow-probability==0.9.0
 numpy>=1.17.0
 matplotlib>=3.1.1
 gym>=0.16.0