ad jaxlib to pyproject

Author: jtgrasb

.github/workflows/pr.yml

@@ -53,8 +53,6 @@ jobs:
         echo "  - python=${{ matrix.python-version }}" >> environment.yml
         echo "  - capytaine" >> environment.yml
         echo "  - wavespectra" >> environment.yml
-        echo "  - jax" >> environment.yml
-        echo "  - jaxlib" >> environment.yml
         cat environment.yml
 
     # use the cache if it exists

pyproject.toml

@@ -22,7 +22,8 @@ dependencies = [
     "numpy>=1.20, <2.0",
     "scipy",
     "xarray",
-    "jax",
+    "jax==0.5.3",
+    "jaxlib==0.5.3",
     "capytaine",
     "joblib",
     "wavespectra>=4.0",