Skip to content

Pull requests: google/orbax

New pull request New
264 Open 2,244 Closed
264 Open 2,244 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

No public description
#2669 opened Dec 6, 2025 by copybara-service bot Loading…
1
Improve type annotations in checkpoint_args.py.
#2668 opened Dec 5, 2025 by copybara-service bot Loading…
Add Orbax Checkpoint V1 documentation section to table of contents.
#2667 opened Dec 5, 2025 by copybara-service bot Loading…
Internal change.
#2666 opened Dec 5, 2025 by copybara-service bot Loading…
1
Internal change.
#2665 opened Dec 5, 2025 by copybara-service bot Loading…
1
Enforce topology check in StandardCheckpointHandler.restore
#2663 opened Dec 5, 2025 by copybara-service bot Loading…
Remove v0 logic out of OrbaxLayout which now only represents v1 checkpoints/layouts
#2662 opened Dec 4, 2025 by copybara-service bot Loading…
Split v0 checkpoint format/layout logic out from OrbaxLayout and into seperate V0Layout
#2661 opened Dec 4, 2025 by copybara-service bot Loading…
Add testing to verify backwards compatibility when missing _CHECKPOINT_METADATA or saving pytree directly to checkpoint.
#2660 opened Dec 4, 2025 by copybara-service bot Loading…
Fix composite handler's extraction logic for handler type strings when missing _CHECKPOINT_METADATA
#2659 opened Dec 4, 2025 by copybara-service bot Loading…
Integrate TPU Inference Converter V2 for bfloat16 optimization in Orbax export.
#2658 opened Dec 4, 2025 by copybara-service bot Loading…
Ensure metadata and loading works correctly when missing _CHECKPOINT_METADATA file.
#2656 opened Dec 4, 2025 by copybara-service bot Loading…
Add logging for the checkpoint config, in addition to benchmark options.
#2655 opened Dec 3, 2025 by copybara-service bot Loading…
Add use_ocdbt property to PyTree metadata.
#2654 opened Dec 3, 2025 by copybara-service bot Loading…
No changes.
#2648 opened Dec 2, 2025 by copybara-service bot Loading…
1
Add select method to PartsOf.
#2647 opened Dec 2, 2025 by copybara-service bot Loading…
1
Internal change
#2646 opened Dec 2, 2025 by copybara-service bot Loading…
1
Temporarily disable proto compilation for orbax/experimental/model in doc serving script.
#2644 opened Dec 2, 2025 by copybara-service bot Loading…
No public description
#2634 opened Nov 26, 2025 by copybara-service bot Loading…
Fix checkpoint key value restoration when working with dictionary integer keys
#2632 opened Nov 26, 2025 by copybara-service bot Loading…
1
Re-enable Orbax Guitar presubmit check.
#2625 opened Nov 25, 2025 by copybara-service bot Loading…
1
Re-enable Orbax Guitar presubmit check.
#2624 opened Nov 25, 2025 by copybara-service bot Loading…
1
Cleanup tmp dirs in background if possible
#2620 opened Nov 24, 2025 by findmyway Loading…
Fix integer dict key restoration in metadata serialization (#2587)
#2619 opened Nov 23, 2025 by Emmanuel-Arokiaraj Loading…
1
1/N DAG orchestration support.
#2618 opened Nov 21, 2025 by copybara-service bot Loading…
ProTip! What’s not been updated in a month: updated:<2025-11-07.