Skip to content

Pull requests: google/orbax

New pull request New
269 Open 2,248 Closed
269 Open 2,248 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

Clarify the docstring of ocp.args.JsonRestore().
#2678 opened Dec 9, 2025 by copybara-service bot Loading…
parent CL, DO NOT SUBMIT
#2677 opened Dec 9, 2025 by copybara-service bot Loading…
Enable bfloat16 optimization for Orbax Export preprocessors.
#2676 opened Dec 9, 2025 by copybara-service bot Loading…
Refactor ServingConfig to use ObmExportOptions dataclass.
#2675 opened Dec 8, 2025 by copybara-service bot Loading…
Update V1 Docs
#2674 opened Dec 8, 2025 by copybara-service bot Loading…
Improve V1 docstrings
#2673 opened Dec 8, 2025 by copybara-service bot Loading…
Remove uses of jax.sharding.PmapSharding.
#2672 opened Dec 8, 2025 by copybara-service bot Loading…
Internal change.
#2671 opened Dec 8, 2025 by copybara-service bot Loading…
1
No public description
#2670 opened Dec 8, 2025 by copybara-service bot Loading…
1
No public description
#2669 opened Dec 6, 2025 by copybara-service bot Loading…
1
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…
No changes.
#2648 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
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.