add jax to environment manually

Author: jtgrasb

.github/workflows/pr.yml

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