Update setup.py

google · Sep 28, 2023 · fc11f58 · fc11f58
1 parent b170aaa
commit fc11f58
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -28,6 +28,7 @@
 INSTALL_REQUIRES = [
     'jax>=0.4.14',
     'frozendict>=2.3.8',
+    'tensorflow>=2.14.0',
     'tf2jax>=0.3.5',
 ]