Bump JAX version

PiperOrigin-RevId: 519041302
google · Apr 19, 2023 · d00a6e0 · d00a6e0
1 parent 49f2c07
commit d00a6e0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/neural_tangents/__init__.py b/neural_tangents/__init__.py
@@ -16,7 +16,7 @@
 """Public Neural Tangents modules and functions."""
 
 
-__version__ = '0.6.2'
+__version__ = '0.6.3'
 
 from . import experimental
 from . import predict

diff --git a/setup.py b/setup.py
@@ -26,7 +26,7 @@
 
 
 INSTALL_REQUIRES = [
-    'jax>=0.4.3',
+    'jax>=0.4.6',
     'frozendict>=2.3',
     'typing_extensions>=4.0.1',
     'tf2jax>=0.3.3',