From a017fabfba7cac85cc88c1e4c02ee9d40bdfcefd Mon Sep 17 00:00:00 2001
From: Roman Novak <romann@google.com>
Date: Mon, 7 Oct 2024 10:54:01 -0700
Subject: [PATCH] Update versions

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 8dfef3e..327cf7f 100644
--- a/setup.py
+++ b/setup.py
@@ -28,7 +28,7 @@
 INSTALL_REQUIRES = [
     'jax>=0.4.34',
     'frozendict>=2.4.5',
-    'tensorflow>=2.17',
+    'tensorflow>=2.16.2',
     'keras>=3.6.0',
     'tf2jax>=0.3.6',
 ]