Skip to content

Pull requests: google/orbax

New pull request New
126 Open 1,238 Closed
126 Open 1,238 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Internal change.
#1483 opened Jan 10, 2025 by copybara-service bot Loading…
Internal change.
#1479 opened Jan 9, 2025 by copybara-service bot Loading…
Add checkpoint/_src/metadata/BUILD.
#1471 opened Jan 8, 2025 by copybara-service bot Loading…
Internal
#1464 opened Jan 6, 2025 by copybara-service bot Loading…
Add _update_metadata() method to CheckpointHandlers.
#1463 opened Jan 5, 2025 by copybara-service bot Loading…
Support custom PyTree metadata.
#1461 opened Jan 3, 2025 by copybara-service bot Loading…
Add .gitignore
#1460 opened Jan 3, 2025 by hartikainen Loading…
Exclude BUILD files from flit builds
#1458 opened Jan 2, 2025 by hartikainen Loading…
1
Save StepMetadata in the AsyncCheckpointer.
#1454 opened Dec 20, 2024 by copybara-service bot Loading…
Save StepMetadata in the Checkpointer.
#1449 opened Dec 18, 2024 by copybara-service bot Loading…
my public commit msg
#1446 opened Dec 17, 2024 by copybara-service bot Loading…
1
Internal change.
#1433 opened Dec 11, 2024 by copybara-service bot Loading…
Add logs in consistent_restore_mesh to examine the invariant.
#1427 opened Dec 5, 2024 by copybara-service bot Loading…
1
Internal
#1417 opened Dec 2, 2024 by copybara-service bot Loading…
1
internal change.
#1412 opened Nov 30, 2024 by copybara-service bot Loading…
Internal
#1406 opened Nov 26, 2024 by copybara-service bot Loading…
Internal
#1393 opened Nov 25, 2024 by copybara-service bot Loading…
Internal
#1390 opened Nov 25, 2024 by copybara-service bot Loading…
Need to add because parent CL updates Orbax codebase.
#1369 opened Nov 21, 2024 by copybara-service bot Loading…
Internal
#1350 opened Nov 18, 2024 by copybara-service bot Loading…
Add MultiProcessTest base.
#1328 opened Nov 13, 2024 by copybara-service bot Loading…
Resubmit single array error message
#1323 opened Nov 12, 2024 by cpgaffney1 Loading…
2
internal change
#1312 opened Nov 8, 2024 by copybara-service bot Loading…
Fix local restore by re-mapping device ids directly instead of inferring them from how process indexes changed across restarts with some false assumptions.
#1310 opened Nov 8, 2024 by copybara-service bot Loading…
1
Fix a bug in Orbax checkpointing where None values PyTree are not handled correctly. This is caused by a recent update in jax where None values are no longer considered as a leaf node: https://jax.readthedocs.io/en/latest/changelog.html#jax-0-4-34-october-4-2024.
#1286 opened Oct 31, 2024 by copybara-service bot Loading…
1
ProTip! Add no:assignee to see everything that’s not assigned.