Update versions

.github/workflows/linux.yml

@@ -32,7 +32,7 @@ jobs:
       uses: styfle/[email protected]
       with:
         access_token: ${{ github.token }}
-    - uses: actions/checkout@v4.1.7
+    - uses: actions/checkout@v4.2.1
 
     - name: Set up Python ${{ matrix.python-version }}
       uses: actions/[email protected]
@@ -54,7 +54,7 @@ jobs:
         JAX_ENABLE_X64=${{ matrix.JAX_ENABLE_X64 }} PYTHONHASHSEED=0 pytest -n auto --cov=neural_tangents --cov-report=xml --cov-report=term
 
     - name: Upload coverage to Codecov
-      uses: codecov/codecov-action@v4.5.0
+      uses: codecov/codecov-action@v4.6.0
       with:
         file: ./coverage.xml
 

.github/workflows/macos.yml

@@ -32,7 +32,7 @@ jobs:
       uses: styfle/[email protected]
       with:
         access_token: ${{ github.token }}
-    - uses: actions/checkout@v4.1.7
+    - uses: actions/checkout@v4.2.1
 
     - name: Set up Python ${{ matrix.python-version }}
       uses: actions/[email protected]
@@ -54,7 +54,7 @@ jobs:
         JAX_ENABLE_X64=${{ matrix.JAX_ENABLE_X64 }} PYTHONHASHSEED=0 pytest -n auto --cov=neural_tangents --cov-report=xml --cov-report=term
 
     - name: Upload coverage to Codecov
-      uses: codecov/codecov-action@v4.5.0
+      uses: codecov/codecov-action@v4.6.0
       with:
         file: ./coverage.xml
 

.github/workflows/pytype.yml

@@ -27,7 +27,7 @@ jobs:
       uses: styfle/[email protected]
       with:
         access_token: ${{ github.token }}
-    - uses: actions/checkout@v4.1.7
+    - uses: actions/checkout@v4.2.1
 
     - name: Set up Python 3.10
       uses: actions/[email protected]

.github/workflows/sketching.yml

@@ -31,7 +31,7 @@ jobs:
       uses: styfle/[email protected]
       with:
         access_token: ${{ github.token }}
-    - uses: actions/checkout@v4.1.7
+    - uses: actions/checkout@v4.2.1
 
     - name: Set up Python ${{ matrix.python-version }}
       uses: actions/[email protected]
@@ -53,7 +53,7 @@ jobs:
         JAX_ENABLE_X64=${{ matrix.JAX_ENABLE_X64 }} PYTHONHASHSEED=0 pytest experimental/tests/ -n auto --cov=experimental/ --cov-report=xml --cov-report=term
 
     - name: Upload coverage to Codecov
-      uses: codecov/codecov-action@v4.5.0
+      uses: codecov/codecov-action@v4.6.0
       with:
         file: ./coverage.xml
 

setup.py

@@ -26,16 +26,16 @@
 
 
 INSTALL_REQUIRES = [
-    'jax>=0.4.32',
-    'frozendict>=2.4.4',
-    'tensorflow>=2.16.2',
-    'keras>=3.5.0',
+    'jax>=0.4.34',
+    'frozendict>=2.4.5',
+    'tensorflow>=2.17',
+    'keras>=3.6.0',
     'tf2jax>=0.3.6',
 ]
 
 
 TESTS_REQUIRES = [
-    'more-itertools>=10.4.0',
+    'more-itertools>=10.5.0',
     'tensorflow-datasets>=4.9.6',
     'flax>=0.9.0',
 ]