bump jax version?

stanford-crfm · Apr 25, 2024 · 6b926fe · 6b926fe
1 parent 776864f
commit 6b926fe
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/.github/workflows/run_entry_tests.yaml b/.github/workflows/run_entry_tests.yaml
@@ -9,7 +9,7 @@ jobs:
     strategy:
       matrix:
         python-version: ["3.10"]
-        jax-version: ["0.4.23"]
+        jax-version: ["0.4.26"]
 
     steps:
       - uses: actions/checkout@v3

diff --git a/.github/workflows/run_ray_tests.yaml b/.github/workflows/run_ray_tests.yaml
@@ -9,7 +9,7 @@ jobs:
     strategy:
       matrix:
         python-version: ["3.10"]
-        jax-version: ["0.4.23"]
+        jax-version: ["0.4.26"]
 
     steps:
       - uses: actions/checkout@v3

diff --git a/.github/workflows/run_tests.yaml b/.github/workflows/run_tests.yaml
@@ -9,7 +9,7 @@ jobs:
     strategy:
       matrix:
         python-version: ["3.10"]
-        jax-version: ["0.4.23"]
+        jax-version: ["0.4.26"]
 
     steps:
       - uses: actions/checkout@v3