Skip to content

Pull requests: google/orbax

New pull request New
154 Open 1,406 Closed
154 Open 1,406 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
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

#v1 Add save/load tests to exercise loading from JAX to numpy and vice versa.
#1689 opened Mar 7, 2025 by copybara-service bot Loading…
Consolidate item handler metadata validation and processing.
#1688 opened Mar 7, 2025 by copybara-service bot Loading…
#v1 Add CheckpointMetadata definitions.
#1687 opened Mar 7, 2025 by copybara-service bot Loading…
#v1 Refactor Context and expand global options.
#1685 opened Mar 7, 2025 by copybara-service bot Loading…
Remove descriptor creation from CompositeCheckpointHandler
#1683 opened Mar 7, 2025 by copybara-service bot Loading…
Enable exporting of batch options in OEX -> OBM.
#1678 opened Mar 6, 2025 by copybara-service bot Loading…
1
Re-enable replica_parallel feature. This was previously disabled due to performance concerns, but these have been identified as relating to the pinned_host_transfer feature instead. Confirmed with a performance test on Gemini XS model, with 16 hosts.
#1672 opened Mar 6, 2025 by copybara-service bot Loading…
Allow configuring the minimum interval seconds for continuous checkpointing.
#1666 opened Mar 6, 2025 by copybara-service bot Loading…
1
Simplify heartbeat options for PjRt distributed runtime.
#1663 opened Mar 5, 2025 by copybara-service bot Loading…
Internal change.
#1662 opened Mar 5, 2025 by copybara-service bot Loading…
Internal
#1661 opened Mar 5, 2025 by copybara-service bot Loading…
1
Enable ArrayMetadata persistence globally.
#1659 opened Mar 5, 2025 by copybara-service bot Loading…
Support minimum interval seconds since last save in Continuous checkpoint save policy, Cloned from CL 732815358.
#1653 opened Mar 4, 2025 by copybara-service bot Loading…
1
Add a _to_jax_shape function to allow partial TF shape to JAX symbolic shape conversion.
#1650 opened Mar 3, 2025 by copybara-service bot Loading…
1
Internal change.
#1649 opened Mar 3, 2025 by copybara-service bot Loading…
Internal change.
#1648 opened Mar 3, 2025 by copybara-service bot Loading…
internal
#1647 opened Mar 3, 2025 by copybara-service bot Loading…
Support minimum_interval_seconds since last save in Continuous checkpoint save policy.
#1645 opened Mar 3, 2025 by copybara-service bot Loading…
1
Enable ArrayMetadataStore by default
#1636 opened Feb 26, 2025 by copybara-service bot Loading…
Internal.
#1634 opened Feb 26, 2025 by copybara-service bot Loading…
Make the create_directories_asynchronously option default to True if jax distributed client is initialized.
#1630 opened Feb 25, 2025 by copybara-service bot Loading…
Internal change.
#1625 opened Feb 21, 2025 by copybara-service bot Loading…
internal
#1620 opened Feb 19, 2025 by copybara-service bot Loading…
internal
#1615 opened Feb 18, 2025 by copybara-service bot Loading…
Add support to create dummy directories in orbax benchmark.
#1613 opened Feb 18, 2025 by copybara-service bot Loading…
ProTip! Follow long discussions with comments:>50.