Skip to content

Pull requests: google/orbax

New pull request New
250 Open 2,228 Closed
250 Open 2,228 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 change
#2637 opened Nov 28, 2025 by copybara-service bot Loading…
1
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
Fix checkpoint key value restoration when working with dictionary integer keys
#2629 opened Nov 25, 2025 by copybara-service bot Loading…
1
Add LOW_PRIORITY_PADDING_WITH_NEXT_ALLOWED_BATCH_SIZE to MixedPriorityBatchingPolicy.
#2627 opened Nov 25, 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
Use ServingConfig.get_signature_keys() to retrieve signature keys.
#2622 opened Nov 24, 2025 by copybara-service bot Loading…
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
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
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
Run all correctness benchmarks in Github
#2603 opened Nov 19, 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…
Always write array metadata if self._array_metadata_store is not None.
#2580 opened Nov 13, 2025 by copybara-service bot Loading…
Internal change.
#2576 opened Nov 12, 2025 by copybara-service bot Loading…
Add checkpoint_manager_test.py to multiprocess tests.
#2547 opened Nov 5, 2025 by copybara-service bot Loading…
1
ProTip! Filter pull requests by the default branch with base:main.