Skip to content

Pull requests: google/orbax

New pull request New
61 Open 717 Closed
61 Open 717 Closed
Author
Filter by author
Label
Filter by label
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Milestones
Filter by milestone
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

Internal change.
#859 opened May 2, 2024 by copybara-service bot Loading…
Add try catch block to catch ValueError reading from sharding string.
#857 opened May 1, 2024 by copybara-service bot Loading…
Factor out core PyTree checkpointing logic into a PyTreeCheckpointHandlerImpl class. PyTreeCheckpointHandler will remain as the entry point for users, and it will keep extra layers of logic dealing with transformations and legacy compatibility (like msgpack checkpoints).
#854 opened Apr 29, 2024 by copybara-service bot Loading…
Add OCDBT support to GlobalHostArray type_handler
#843 opened Apr 24, 2024 by copybara-service bot Loading…
Modify JsonCheckpointHandler to Async.
#824 opened Apr 16, 2024 by copybara-service bot Loading…
Add an option to allow sharding consolidation / reduction to work around the dtensor limitation of disallowing one dimension being sharded across multiple axis names.
#822 opened Apr 15, 2024 by copybara-service bot Loading…
1
Create the utils function load_model_with_tfrt API for XLA devices.
#813 opened Apr 12, 2024 by copybara-service bot Loading…
Internal change.
#801 opened Apr 6, 2024 by copybara-service bot Loading…
Internal change.
#800 opened Apr 6, 2024 by copybara-service bot Loading…
Enable Orbax checkpointing with background deleting thread
#796 opened Apr 4, 2024 by copybara-service bot Loading…
remove all_params_aggregated.
#794 opened Apr 3, 2024 by copybara-service bot Loading…
Internal
#783 opened Apr 2, 2024 by copybara-service bot Loading…
1
Create an interface for retrieving the current step number.
#778 opened Mar 30, 2024 by copybara-service bot Loading…
1
Internal change.
#754 opened Mar 15, 2024 by copybara-service bot Loading…
file access hardening on model exporter.
#717 opened Feb 14, 2024 by copybara-service bot Loading…
1
Changes default timeout in AsyncCheckpointer.
#698 opened Feb 1, 2024 by copybara-service bot Loading…
Orbax test. Used for testing, don't submit.
#694 opened Jan 29, 2024 by copybara-service bot Loading…
1
Fix errors of loading ckpt that contains sharding configs.
#671 opened Jan 17, 2024 by copybara-service bot Loading…
1
orbax change is from diffbase
#669 opened Jan 17, 2024 by copybara-service bot Loading…
1
Provide better support for custom CheckpointHandlers without registered CheckpointArgs by providing a wrapper CheckpointHandler as a fallback. This class is introduced for backwards compatibility, and will eventually be removed.
#618 opened Dec 7, 2023 by copybara-service bot Loading…
Replace references to deprecated device_buffer attributes
#616 opened Dec 6, 2023 by copybara-service bot Loading…
Internal.
#597 opened Nov 23, 2023 by copybara-service bot Loading…
The internal changes.
#596 opened Nov 20, 2023 by copybara-service bot Loading…
1
Internal changes to support more export / load options.
#593 opened Nov 17, 2023 by copybara-service bot Loading…
1
Internal change.
#592 opened Nov 15, 2023 by copybara-service bot Loading…
ProTip! What’s not been updated in a month: updated:<2024-04-03.