Skip to content

Add elastic pause/resume functionality to MaxText.#3518

Merged
copybara-service[bot] merged 1 commit intomainfrom
test_890594825
Apr 7, 2026
Merged

Add elastic pause/resume functionality to MaxText.#3518
copybara-service[bot] merged 1 commit intomainfrom
test_890594825

Conversation

@copybara-service
Copy link
Copy Markdown
Contributor

@copybara-service copybara-service bot commented Mar 30, 2026

Add elastic pause/resume functionality to MaxText.

@copybara-service copybara-service bot force-pushed the test_890594825 branch 2 times, most recently from 53a3aa8 to c639168 Compare March 30, 2026 03:55
@codecov
Copy link
Copy Markdown

codecov bot commented Mar 30, 2026 •
edited
Loading

Codecov Report

❌ Patch coverage is 50.00000% with 40 lines in your changes missing coverage. Please review.

Files with missing lines Patch % Lines
src/maxtext/utils/gcs_utils.py 11.11% 16 Missing ⚠️
src/maxtext/utils/elastic_utils.py 73.46% 11 Missing and 2 partials ⚠️
src/maxtext/trainers/pre_train/train.py 15.38% 11 Missing ⚠️

📢 Thoughts on this report? Let us know!

@copybara-service copybara-service bot force-pushed the test_890594825 branch 7 times, most recently from 465ac5c to f2b2227 Compare March 30, 2026 05:45
@copybara-service copybara-service bot force-pushed the test_890594825 branch 26 times, most recently from ffa2c29 to aeb99f6 Compare April 7, 2026 03:30
@copybara-service copybara-service bot merged commit 1963688 into main Apr 7, 2026
@copybara-service copybara-service bot deleted the test_890594825 branch April 7, 2026 03:41
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@gobbleturk gobbleturk Awaiting requested review from gobbleturk gobbleturk is a code owner

@khatwanimohit khatwanimohit Awaiting requested review from khatwanimohit khatwanimohit is a code owner

@bvandermoon bvandermoon Awaiting requested review from bvandermoon bvandermoon 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

@SurbhiJainUSC SurbhiJainUSC Awaiting requested review from SurbhiJainUSC SurbhiJainUSC 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 jesselu-google is a code owner

@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

Assignees

@abhinavclemson abhinavclemson

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

1 participant

@abhinavclemson