diff --git a/setup.py b/setup.py
index 7f9a4bd6..dccec278 100644
--- a/setup.py
+++ b/setup.py
@@ -28,9 +28,9 @@
     install_requires=[
         "numpy",
         "dm_tree",
-        "tensorflow[and-cuda]==2.16.1.*",
+        "tensorflow[and-cuda]==2.15.1.*",
         "tensorflow_io",
-        "tensorflow-probability[tf]==0.24.0",
+        "tensorflow-probability[tf]==0.23.0",
         "dm_sonnet",
         "wandb",
         "absl-py",