Update versions

romanngg · romanngg · commit 2da21b1072dc · 2024-10-24T16:14:31.000-07:00
diff --git a/setup.py b/setup.py
@@ -28,7 +28,7 @@
 INSTALL_REQUIRES = [
     'jax>=0.4.34',
     'frozendict>=2.4.6',
-    'tensorflow>=2.16.2',
+    'tensorflow>=2.17.1',
     'keras>=3.6.0',
     'tf2jax>=0.3.6',
 ]
