JAX version is 0.4.35 until Diffrax new release

DifferentiableUniverseInitiative · Dec 8, 2024 · b264da5 · b264da5
1 parent d8c68ac
commit b264da5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -29,7 +29,7 @@ jobs:
       run: |
         sudo apt-get install -y libopenmpi-dev
         python -m pip install --upgrade pip
-        pip install jax=0.4.35
+        pip install jax==0.4.35
         pip install .[test]
 
     - name: Run Single Device Tests