diff --git a/requirements.txt b/requirements.txt
index 0d13788..5da6ba0 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -7,7 +7,7 @@ pytest-cov
 coveralls
 autograd
 tensorflow>=2
-tensorflow-probability
+tensorflow-probability[tf]
 torch
 jax
 jaxlib