Move examples under maxtext/examples

[NuojCheng]

Description

This PR consolidates and reorganizes the example notebooks and pedagogical scripts within the repository. Specifically, it moves files from src/MaxText/examples/ and pedagogical_examples/ into a unified src/maxtext/examples/ directory. All internal references, documentation links, and CI workflows have been updated to reflect these path changes.

Key Changes

• Directory Consolidation:
  • Moved all Jupyter notebooks (e.g., rl_llama3_demo.ipynb, sft_llama3_demo.ipynb, demo_decoding.ipynb) from src/MaxText/examples/ to src/maxtext/examples/.
  • Moved pedagogical scripts (e.g., non_spmd.py, shardings.py, shmap_collective_matmul.py) from the top-level pedagogical_examples/ directory to src/maxtext/examples/.
• Workflow Updates: Updated .github/workflows/run_jupyter_notebooks.yml to set the new MAXTEXT_NOTEBOOKS_ROOT path and include src in PYTHONPATH.
• Documentation & Links:
  • Updated README.md and various files in docs/ to ensure links point to the new examples directory.
  • Updated notebook metadata and internal code blocks that referenced the old paths.
• Configuration & Tests:
  • Updated rl.yml configuration to point to the new chat_template_path.
  • Updated integration tests to import from the new location.
  • Updated codecov.yml to reflect the new directory structure for coverage ignore rules.

Tests

CI

Checklist

Before submitting this PR, please make sure (put X in square brackets):

• I have performed a self-review of my code. For an optional AI review, add the gemini-review label.
• I have necessary comments in my code, particularly in hard-to-understand areas.
• I have run end-to-end tests tests and provided workload links above if applicable.
• I have made or will make corresponding changes to the doc if needed, including adding new documentation pages to the relevant Table of Contents (toctree directive) as explained in our documentation.

[codecov]

Codecov Report

✅ All modified and coverable lines are covered by tests.

📢 Thoughts on this report? Let us know!

[khatwanimohit]

Great catch! :)

[khatwanimohit]

lgtm