Skip to content

Update Gemma3-4b E2E post-training scripts for pathways#4168

Merged
copybara-service[bot] merged 1 commit into
mainfrom
post_train_gemma3_4b
Jun 15, 2026
Merged

Update Gemma3-4b E2E post-training scripts for pathways#4168
copybara-service[bot] merged 1 commit into
mainfrom
post_train_gemma3_4b

Conversation

@SurbhiJainUSC

@SurbhiJainUSC SurbhiJainUSC commented Jun 15, 2026 •
edited
Loading

Copy link
Copy Markdown
Collaborator

Description

This PR updates the E2E post-training scripts for Gemma3-4B to fully support Pathways. It ensures that the scripts are compatible with Pathways by explicitly setting use_pathways=True.

Tests

Tested on pathways GKE cluster.

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 Bot commented Jun 15, 2026

Copy link
Copy Markdown

Codecov Report

✅ All modified and coverable lines are covered by tests.

📢 Thoughts on this report? Let us know!

@SurbhiJainUSC SurbhiJainUSC marked this pull request as ready for review June 15, 2026 21:27
YixuanWang-99
YixuanWang-99 approved these changes Jun 15, 2026
View reviewed changes
Comment thread tests/end_to_end/tpu/gemma3/4b/test_gemma3_rl.sh
@copybara-service copybara-service Bot merged commit e0b220a into main Jun 15, 2026
76 checks passed
@copybara-service copybara-service Bot deleted the post_train_gemma3_4b branch June 15, 2026 23:00
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@bvandermoon bvandermoon bvandermoon approved these changes
@khatwanimohit khatwanimohit khatwanimohit approved these changes
@YixuanWang-99 YixuanWang-99 YixuanWang-99 approved these changes
@abhinavclemson abhinavclemson Awaiting requested review from abhinavclemson abhinavclemson is a code owner
@gobbleturk gobbleturk Awaiting requested review from gobbleturk gobbleturk is a code owner
@vipannalla vipannalla Awaiting requested review from vipannalla vipannalla is a code owner
@RissyRan RissyRan Awaiting requested review from RissyRan RissyRan is a code owner
@richjames0 richjames0 Awaiting requested review from richjames0 richjames0 is a code owner
@gagika gagika Awaiting requested review from gagika gagika is a code owner
@shralex shralex Awaiting requested review from shralex shralex is a code owner
@hengtaoguo hengtaoguo Awaiting requested review from hengtaoguo hengtaoguo is a code owner
@A9isha A9isha Awaiting requested review from A9isha A9isha is a code owner
@aireenmei aireenmei Awaiting requested review from aireenmei aireenmei is a code owner
@NuojCheng NuojCheng Awaiting requested review from NuojCheng NuojCheng is a code owner
@jiangjy1982 jiangjy1982 Awaiting requested review from jiangjy1982 jiangjy1982 is a code owner
@suexu1025 suexu1025 Awaiting requested review from suexu1025 suexu1025 is a code owner
@NicoGrande NicoGrande Awaiting requested review from NicoGrande NicoGrande is a code owner
@jesselu-google jesselu-google Awaiting requested review from jesselu-google
@dipannita08 dipannita08 Awaiting requested review from dipannita08 dipannita08 is a code owner
@igorts-git igorts-git Awaiting requested review from igorts-git igorts-git is a code owner
@darisoy darisoy Awaiting requested review from darisoy darisoy is a code owner

Assignees

No one assigned

Labels

pull ready

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

4 participants

@SurbhiJainUSC @bvandermoon @khatwanimohit @YixuanWang-99