fix jax and jaxlib version 0.4.28

qutip · May 12, 2024 · 18d61e7 · 18d61e7
1 parent 9eb39ee
commit 18d61e7
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -65,7 +65,7 @@ jobs:
 
       - name: Install fixed version jax and jaxlib
         run: |
-          python -m pip install 'jax==0.4.23' 'jaxlib==0.4.23'
+          python -m pip install 'jax==0.4.28' 'jaxlib==0.4.28'
 
       - name: Install qutip-jax from GitHub
         if: ${{ matrix.qutip-version == '@master'}}

diff --git a/doc/requirements.txt b/doc/requirements.txt
@@ -1,7 +1,7 @@
 numpy>=1.16.6
 scipy>=1.10.1
-jax>=0.4.23
-jaxlib>=0.4.23
+jax==0.4.28
+jaxlib==0.4.28
 qutip>=5.0.1
 cython>=0.29.33
 numpydoc==1.5.0

diff --git a/doc/rtd-environment.yml b/doc/rtd-environment.yml
@@ -4,8 +4,8 @@ channels:
 dependencies:
   - numpy>=1.16.6
   - scipy>=1.10.1
-  - jax>=0.4.23
-  - jaxlib>=0.4.23
+  - jax==0.4.28
+  - jaxlib==0.4.28
   - cython>=0.29.33
   - sphinx==6.1.3
   - sphinx_rtd_theme==1.2.0