Skip to content

Pull requests: google/orbax

New pull request New
228 Open 2,007 Closed
228 Open 2,007 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
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Fix _glob_step_paths if HNS is enabled
#2384 opened Sep 13, 2025 by dvruette Loading…
1
Expose merge_trees in orbax.checkpoint.experimental.v1.tree.
#2383 opened Sep 12, 2025 by copybara-service bot Loading…
Internal change.
#2382 opened Sep 12, 2025 by copybara-service bot Loading…
Add arrays submodule to Orbax Checkpoint experimental v1.
#2381 opened Sep 12, 2025 by copybara-service bot Loading…
Change ValueError to NotImplementedError for unsupported partial saving replacement.
#2379 opened Sep 12, 2025 by copybara-service bot Loading…
: #v1 Modify LeafHandler definitions so that AbstractLeaf or Type[AbstractLeaf] are always accepted as valid abstract values.
#2378 opened Sep 12, 2025 by copybara-service bot Loading…
Migrate Tunelab to Orbax PreservationPolicy for checkpoint management.
#2376 opened Sep 12, 2025 by copybara-service bot Loading…
Make PyTreeCheckpointHandlerTest a single test file.
#2375 opened Sep 12, 2025 by copybara-service bot Loading…
Delete old tf related functions from OEX.
#2374 opened Sep 11, 2025 by copybara-service bot Loading…
Add a guide for Partial Saving.
#2373 opened Sep 11, 2025 by copybara-service bot Loading…
Update EveryNSteps preservation policy.
#2366 opened Sep 10, 2025 by copybara-service bot Loading…
#v1 Adjust leaf and abstract leaf definitions.
#2365 opened Sep 9, 2025 by copybara-service bot Loading…
#v1 Add guide for Orbax Checkpointing for PyTorch Users.
#2363 opened Sep 9, 2025 by copybara-service bot Loading…
1
Add PreservationPolicy support to EmergencyCheckpointManager.
#2355 opened Sep 8, 2025 by copybara-service bot Loading…
1
access state from restored composite args after restore
#2353 opened Sep 8, 2025 by keshavb96 Loading…
Internal change
#2351 opened Sep 5, 2025 by copybara-service bot Loading…
Internal change
#2350 opened Sep 5, 2025 by copybara-service bot Loading…
Propagate batch_padding_policy through BatchOptions in JSV
#2346 opened Sep 4, 2025 by copybara-service bot Loading…
1
Internal change
#2343 opened Sep 4, 2025 by copybara-service bot Loading…
Use TfDataProcessor in OEX
#2337 opened Sep 3, 2025 by copybara-service bot Loading…
Update sidv2, yt_prod_v3 test data and re-enable the test.
#2335 opened Sep 3, 2025 by copybara-service bot Loading…
Add workflow to test running multihost benchmark workflow
#2328 opened Sep 2, 2025 by copybara-service bot Loading…
Adding Orbax PreservationPolicy for checkpoint in GMAX PROD.
#2327 opened Sep 1, 2025 by copybara-service bot Loading…
small changes
#2326 opened Sep 1, 2025 by copybara-service bot Loading…
1
Refactor some test utils to test save and restore operations in more scenarios
#2307 opened Aug 27, 2025 by copybara-service bot Loading…
1
ProTip! Adding no:label will show everything without a label.