Update setup.py

google · Nov 27, 2022 · 0d8dea2 · 0d8dea2
1 parent 98bb23b
commit 0d8dea2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -26,10 +26,10 @@
 
 
 INSTALL_REQUIRES = [
-    'jax>=0.3.13',
+    'jax>=0.3.14',
     'frozendict>=2.3',
     'typing_extensions>=4.0.1',
-    'tf2jax>=0.3.0',
+    'tf2jax>=0.3.1',
 ]