MAINT: add jax[cpu] to execution checks (#260)

Author: mmcky

.github/workflows/execution-linux.yml

@@ -24,6 +24,10 @@ jobs:
         run: |
           conda install anaconda
           pip install jupyter-book sphinx-multitoc-numbering quantecon-book-theme sphinxext-rediraffe sphinx-tojupyter sphinx-exercise sphinx-togglebutton
+      - name: Install Jax [CPU]
+        shell: bash -l {0}
+        run: |
+          pip install "jax[CPU]"
       - name: Build Lectures (+ Execution Checks)
         shell: bash -l {0}
         run: jb build lectures --path-output=./ -W --keep-going

.github/workflows/execution-osx.yml

@@ -24,6 +24,10 @@ jobs:
         run: |
           conda install anaconda
           pip install jupyter-book sphinx-multitoc-numbering quantecon-book-theme sphinxext-rediraffe sphinx-tojupyter sphinx-exercise sphinx-togglebutton
+      - name: Install Jax [CPU]
+        shell: bash -l {0}
+        run: |
+          pip install "jax[CPU]"
       - name: Build Lectures (+ Execution Checks)
         shell: bash -l {0}
         run: jb build lectures --path-output=./ -W --keep-going

.github/workflows/execution-win.yml

@@ -29,6 +29,10 @@ jobs:
           conda install -c numba llvmlite
           pip install jupyter-book
           pip install jupyter-book sphinx-multitoc-numbering quantecon-book-theme sphinxext-rediraffe sphinx-tojupyter sphinx-exercise sphinx-togglebutton
+      - name: Install Jax [CPU]
+        shell: bash -l {0}
+        run: |
+          pip install "jax[CPU]"
       - name: Build Lectures (+ Execution Checks)
         shell: powershell
         run: jb build lectures --path-output=./ -W --keep-going