Skip to content

Pull requests: google/orbax

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

Internal changes
#2702 opened Dec 13, 2025 by copybara-service bot Loading…
Move read-time dtype casting into ArrayReadSpec.
#2701 opened Dec 12, 2025 by copybara-service bot Loading…
TODO
#2700 opened Dec 12, 2025 by copybara-service bot Loading…
1
Fix the logic of oex.utils.get_lowering_platforms() as it didn't lower the passed native_serialization_platforms.
#2699 opened Dec 12, 2025 by copybara-service bot Loading…
Refactor jax2obm configuration using a dataclass.
#2698 opened Dec 12, 2025 by copybara-service bot Loading…
Refine bfloat16 conversion in Orbax Export utils.
#2697 opened Dec 12, 2025 by copybara-service bot Loading…
Internal changes
#2694 opened Dec 10, 2025 by copybara-service bot Loading…
#v1 Modify background thread execution to allow the runner to accept multiple concurrent tasks for the background event loop. This allows scheduling multiple operations, like async directory creation and background commits, on the same event loop.
#2686 opened Dec 10, 2025 by copybara-service bot Loading…
parent CL, DO NOT SUBMIT
#2677 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…
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
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…
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…
ProTip! no:milestone will show everything without a milestone.