Skip to content

Pull requests: google/orbax

New pull request New
Clear current search query, filters, and sorts
122 Open 1,203 Closed
122 Open 1,203 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

Expand StepMetadata.item_metadata type to include non-composite cases.
#1442 opened Dec 12, 2024 by copybara-service bot Loading… updated Dec 13, 2024
Internal
#1440 opened Dec 12, 2024 by copybara-service bot Loading… updated Dec 13, 2024
Expand StepMetadata.item_handlers type to include non-composite cases.
#1441 opened Dec 12, 2024 by copybara-service bot Loading… updated Dec 12, 2024
Internal
#1406 opened Nov 26, 2024 by copybara-service bot Loading… updated Dec 12, 2024
Internal change.
#1439 opened Dec 12, 2024 by copybara-service bot Loading… updated Dec 12, 2024
Internal change.
#1432 opened Dec 10, 2024 by copybara-service bot Loading… updated Dec 12, 2024
Internal change.
#1433 opened Dec 11, 2024 by copybara-service bot Loading… updated Dec 11, 2024
Add logs in consistent_restore_mesh to examine the invariant.
#1427 opened Dec 5, 2024 by copybara-service bot Loading… updated Dec 5, 2024
1
Internal
#1417 opened Dec 2, 2024 by copybara-service bot Loading… updated Dec 3, 2024
1
internal change.
#1412 opened Nov 30, 2024 by copybara-service bot Loading… updated Nov 30, 2024
Internal
#1393 opened Nov 25, 2024 by copybara-service bot Loading… updated Nov 25, 2024
Add user-facing TreeMetadata object returned by PyTreeCheckpointHandler.metadata. This object mimics an ordinary PyTree to make the change unnoticeable to most users, but also has additional accessible properties not included in any tree mapping operations.
#1276 opened Oct 28, 2024 by copybara-service bot Loading… updated Nov 25, 2024
Internal
#1390 opened Nov 25, 2024 by copybara-service bot Loading… updated Nov 25, 2024
Need to add because parent CL updates Orbax codebase.
#1369 opened Nov 21, 2024 by copybara-service bot Loading… updated Nov 21, 2024
Internal
#1350 opened Nov 18, 2024 by copybara-service bot Loading… updated Nov 18, 2024
Add MultiProcessTest base.
#1328 opened Nov 13, 2024 by copybara-service bot Loading… updated Nov 14, 2024
Resubmit single array error message
#1323 opened Nov 12, 2024 by cpgaffney1 Loading… updated Nov 12, 2024
internal change
#1312 opened Nov 8, 2024 by copybara-service bot Loading… updated Nov 8, 2024
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… updated Nov 8, 2024
1
Add emergency checkpoint logging of arrays for debugging.
#1198 opened Sep 23, 2024 by copybara-service bot Loading… updated Nov 7, 2024
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… updated Oct 31, 2024
1
Internal change.
#1282 opened Oct 30, 2024 by copybara-service bot Loading… updated Oct 30, 2024
Add shutdown logging wrapper.
#1263 opened Oct 21, 2024 by copybara-service bot Loading… updated Oct 22, 2024
Update fix:
#1260 opened Oct 21, 2024 by lamcodeofpwnosec Loading… updated Oct 21, 2024
1
Internal.
#1233 opened Oct 8, 2024 by copybara-service bot Loading… updated Oct 9, 2024
ProTip! no:milestone will show everything without a milestone.