Skip to content

adjust copybara rules and remove jax.device assertion #3056

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom
Merged
SurbhiJainUSC merged 1 commit into from
Jan 31, 2026
Merged

adjust copybara rules and remove jax.device assertion #3056

SurbhiJainUSC merged 1 commit into from
Jan 31, 2026
+0 −1

Conversation

@copybara-service
Copy link

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

adjust copybara rules and remove jax.device assertion

FUTURE_COPYBARA_INTEGRATE_REVIEW=#2999 from AI-Hypercomputer:rbierneni-qwen3-next-tflops 3fcbcc7

@copybara-service copybara-service bot force-pushed the test_863413100 branch 2 times, most recently from bfbd416 to 7186bab Compare January 30, 2026 23:21
@codecov
Copy link

codecov bot commented Jan 30, 2026

Codecov Report

❌ Patch coverage is 94.59459% with 2 lines in your changes missing coverage. Please review.

Files with missing lines Patch % Lines
src/maxtext/utils/maxtext_utils.py 94.59% 1 Missing and 1 partial ⚠️

📢 Thoughts on this report? Let us know!

@SurbhiJainUSC SurbhiJainUSC merged commit e900d8d into main Jan 31, 2026
@SurbhiJainUSC SurbhiJainUSC deleted the test_863413100 branch January 31, 2026 06: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

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

@parambole parambole Awaiting requested review from parambole

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

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

@jacoguzo jacoguzo Awaiting requested review from jacoguzo

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

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

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

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

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

@shuningjin shuningjin Awaiting requested review from shuningjin

@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

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

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

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

2 participants

@SurbhiJainUSC @Google-ML-Automation