Skip to content

Pull requests: google/orbax

New pull request New
257 Open 2,206 Closed
257 Open 2,206 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

Use ServingConfig.get_signature_keys() to retrieve signature keys.
#2622 opened Nov 24, 2025 by copybara-service bot Loading…
Remove assertions on individual save times in CheckpointManager perf benchmark.
#2621 opened Nov 24, 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
Change the structure of ObmExport Class for go/oex-jsv-graph-orchestration
#2618 opened Nov 21, 2025 by copybara-service bot Loading…
Internal Change
#2617 opened Nov 21, 2025 by copybara-service bot Loading…
Introduce ArrayReadSpec which replaces get_json_tspec_read usage and mirrors the structure of ArrayWriteSpec
#2616 opened Nov 21, 2025 by copybara-service bot Loading…
1
Fix using jax.eval_shape output as StandardRestore.item
#2615 opened Nov 21, 2025 by copybara-service bot Loading…
[3/N] Refactor ObmExport class.
#2614 opened Nov 21, 2025 by copybara-service bot Loading…
Introduce ArrayReadSpec which replaces get_json_tspec_read usage and mirrors the structure of ArrayWriteSpec
#2613 opened Nov 21, 2025 by copybara-service bot Loading…
1
Introduce ArrayReadSpec which replaces get_json_tspec_read usage and mirrors the structure of ArrayWriteSpec
#2612 opened Nov 21, 2025 by copybara-service bot Loading…
1
Introduce ArrayReadSpec which replaces get_json_tspec_read usage and mirrors the structure of ArrayWriteSpec
#2611 opened Nov 21, 2025 by copybara-service bot Loading…
1
Re-enable Orbax Guitar presubmit check.
#2610 opened Nov 21, 2025 by copybara-service bot Loading…
1
Introduce ArrayReadSpec which replaces get_json_tspec_read usage and mirrors the structure of ArrayWriteSpec
#2609 opened Nov 21, 2025 by copybara-service bot Loading…
1
Add priority batch support - Add MixedPriorityBatchingPolicy
#2608 opened Nov 21, 2025 by copybara-service bot Loading…
1
Rename strip_xla_flags to persist_xla_flags and add validation.
#2607 opened Nov 20, 2025 by copybara-service bot Loading…
Internal
#2606 opened Nov 20, 2025 by copybara-service bot Loading…
1
Fixed Copybara issue for Orbax
#2605 opened Nov 20, 2025 by copybara-service bot Loading…
1
Add option to persist_xla_flags from Orbax Native Models.
#2604 opened Nov 19, 2025 by copybara-service bot Loading…
Run all correctness benchmarks in Github
#2603 opened Nov 19, 2025 by copybara-service bot Loading…
1
This PR decouples Orbax from HNS rename API dependencies by delegating the filesystem operations to the underlying TensorFlow gfile implementation. This change improves testability and makes Orbax less susceptible to regressions from this api client changes.
#2600 opened Nov 18, 2025 by copybara-service bot Loading…
1
Add utility for grouping leaf information for partial loading.
#2599 opened Nov 18, 2025 by copybara-service bot Loading…
Update internally to reflect https://github.com/google/orbax/pull/2597
#2598 opened Nov 18, 2025 by copybara-service bot Loading…
Add NumpyLayout for loading NumPy checkpoints.
#2596 opened Nov 18, 2025 by copybara-service bot Loading…
Add a function to group merged metadata leaves by checkpoint index.
#2594 opened Nov 18, 2025 by copybara-service bot Loading…
ProTip! Exclude everything labeled bug with -label:bug.