added jax install to workflow

.github/workflows/ci.yml

@@ -26,6 +26,7 @@ jobs:
         run: |
           python -m pip install --upgrade pip setuptools
           pip install flake8 codecov pytest-cov wheel matplotlib seaborn mne
+          pip install jax[cpu]==0.3.25 -f https://storage.googleapis.com/jax-releases/jax_releases.html
           if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
           pip install .
       - name: Lint with flake8 🎓