Skip to content
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

Add example of unbox() and replace_boxed() to the jit guide #3843

Merged
merged 1 commit into from
Apr 15, 2024

Conversation

IvyZX
Copy link
Collaborator

@IvyZX IvyZX commented Apr 11, 2024

Fixes #3787

Copy link

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB

@IvyZX IvyZX requested review from chiamp and cgarciae April 11, 2024 18:59
@copybara-service copybara-service bot merged commit 9c008e3 into google:main Apr 15, 2024
19 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Orbax checkpoint for LogicallyPartitioned params
2 participants