Skip to content

Pull requests: google/orbax

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

Restruct 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…
Fix checkpoint key value restoration when working with dictionary integer keys
#2592 opened Nov 17, 2025 by copybara-service bot Loading…
1
#v1 Eliminate explicit usages of epath.Path. Use more generic pathlib.Path for type hinting, which will default to epath.Path as the implementation. This will enable more dynamic testing.
#2591 opened Nov 17, 2025 by copybara-service bot Loading…
Add bfloat16 flags and support bfloat16 optimization for native JAX function.
#2584 opened Nov 13, 2025 by copybara-service bot Loading…
Introduce ArrayReadSpec which replaces get_json_tspec_read usage and mirrors the structure of ArrayWriteSpec
#2583 opened Nov 13, 2025 by copybara-service bot Loading…
1
ProTip! Add no:assignee to see everything that’s not assigned.