Skip to content

Migrate Tunix's usage of Orbax V0 Checkpoint Manager to V1 Checkpointer and. migrate distill_utils to v1 orbax semantics.#4116

Open
copybara-service[bot] wants to merge 1 commit into
mainfrom
test_929057022
Open

Migrate Tunix's usage of Orbax V0 Checkpoint Manager to V1 Checkpointer and. migrate distill_utils to v1 orbax semantics.#4116
copybara-service[bot] wants to merge 1 commit into
mainfrom
test_929057022

Conversation

@copybara-service

Copy link
Copy Markdown
Contributor

Migrate Tunix's usage of Orbax V0 Checkpoint Manager to V1 Checkpointer and. migrate distill_utils to v1 orbax semantics.

@codecov

codecov Bot commented Jun 9, 2026

Copy link
Copy Markdown

Codecov Report

❌ Patch coverage is 27.50000% with 29 lines in your changes missing coverage. Please review.

Files with missing lines Patch % Lines
...ners/post_train/distillation/distillation_utils.py 7.14% 13 Missing ⚠️
src/maxtext/common/checkpointing.py 33.33% 12 Missing ⚠️
.../trainers/post_train/distillation/train_distill.py 50.00% 4 Missing ⚠️

📢 Thoughts on this report? Let us know!

…er and. migrate distill_utils to v1 orbax semantics.

PiperOrigin-RevId: 929057022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant