Skip to content

Document all checkpoint policies in one place, on the JAX public API page. #30177

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.

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

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

carlosgmartin
Copy link
Contributor

@carlosgmartin carlosgmartin commented Jul 12, 2025

Document all checkpoint policies (separately listed here and here, currently) in one place: The JAX public API page.

Fixes #12417.

@jakevdp jakevdp self-assigned this Jul 14, 2025
@jakevdp jakevdp self-requested a review July 14, 2025 16:04
Copy link
Collaborator

@jakevdp jakevdp left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks good, thanks! One comment below. Also requesting additional review of the checkpoint policy descriptions from @yashk2810

@jakevdp jakevdp requested a review from yashk2810 July 14, 2025 16:08
@google-ml-butler google-ml-butler bot added kokoro:force-run pull ready Ready for copybara import and testing labels Jul 14, 2025
@jakevdp jakevdp self-requested a review July 14, 2025 16:20
@carlosgmartin carlosgmartin force-pushed the document_checkpoint_policies branch from 5ba45bb to c2b1633 Compare July 14, 2025 18:17
@carlosgmartin carlosgmartin force-pushed the document_checkpoint_policies branch from c2b1633 to 1319253 Compare July 14, 2025 20:46
@carlosgmartin carlosgmartin requested a review from jakevdp July 14, 2025 20:46
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
documentation kokoro:force-run pull ready Ready for copybara import and testing
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Hard to find documentation of predefined jax checkpoint policies
3 participants