add math-verify for run_jupyter_notebooks.yml

Author: A9isha

.github/workflows/run_jupyter_notebooks.yml

@@ -83,6 +83,7 @@ jobs:
 
           uv pip install --no-deps qwix==0.1.4
           uv pip install --no-deps protobuf==5.29.5
+          uv pip install math-verify==0.9.0
           python3 -m pip freeze
       - name: Run Post-Training Notebooks
         shell: bash

src/maxtext/examples/rl_llama3_demo.ipynb

@@ -8,7 +8,7 @@
     "\n",
     "This notebook demonstrates training on Llama3.1-8B-Instruct model with either GRPO (Group Relative Policy Optimization) or GSPO (Group Sequence Policy Optimization).\n",
     "\n",
-    "This notebook can run on **TPU v5e-8** or **v5p-8**.\n",
+    "This notebook can run on **TPU v6e-8** or **v5p-8**.\n",
     "\n",
     "## What is GRPO/GSPO?\n",
     "\n",
@@ -32,7 +32,7 @@
     "1.  Navigate to the menu at the top of the screen.\n",
     "2.  Click on **Runtime**.\n",
     "3.  Select **Change runtime type** from the dropdown menu.\n",
-    "4.  Select **v5e-8** or **v5p-8 TPU** as the **Hardware accelerator**.\n",
+    "4.  Select **v6e-8** or **v5p-8 TPU** as the **Hardware accelerator**.\n",
     "5. Click on **Save**."
    ]
   },

src/maxtext/examples/sft_llama3_demo.ipynb

@@ -17,7 +17,7 @@
     "\n",
     "This notebook demonstrates how to perform Supervised Fine-Tuning (SFT) on Llama3.1-8B-Instruct using the Hugging Face ultrachat_200k dataset with MaxText and Tunix integration for efficient training.\n",
     "\n",
-    "This notebook can run on **TPU v5e-8** or **v5p-8**."
+    "This notebook can run on **TPU v6e-8** or **v5p-8**."
    ]
   },
   {
@@ -32,7 +32,7 @@
     "1.  Navigate to the menu at the top of the screen.\n",
     "2.  Click on **Runtime**.\n",
     "3.  Select **Change runtime type** from the dropdown menu.\n",
-    "4.  Select **v5e-8** or **v5p-8 TPU** as the **Hardware accelerator**.\n",
+    "4.  Select **v6e-8** or **v5p-8 TPU** as the **Hardware accelerator**.\n",
     "5. Click on **Save**."
    ]
   },