-
Notifications
You must be signed in to change notification settings - Fork 35
Pull requests: google/orbax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Introduce
CheckpointManagerOptions.should_keep_fn as an alternative to Introduce CheckpointManagerOptions.keep_period.
#1302
opened Nov 6, 2024 by
copybara-service
bot
Loading…
Fix emergency checkpointing issue arising when repeatedly restoring from a local checkpoint. Process ID remapping may happen in different ways on subsequent times, so we need to ensure new process metadata is saved with every checkpoint, so we can recover the process ID mapping used to save that checkpoint.
#1301
opened Nov 6, 2024 by
copybara-service
bot
Loading…
Make
step.step_metadata_of_checkpoint_path() concurrent to improve performance.
#1296
opened Nov 2, 2024 by
copybara-service
bot
Loading…
Add user-facing CheckpointMetadata classes and Internal classes for IO.
#1290
opened Oct 31, 2024 by
copybara-service
bot
Loading…
Fix a bug in Orbax checkpointing where None values PyTree are not handled correctly. This is caused by a recent update in jax where None values are no longer considered as a leaf node: https://jax.readthedocs.io/en/latest/changelog.html#jax-0-4-34-october-4-2024.
#1286
opened Oct 31, 2024 by
copybara-service
bot
Loading…
Add user-facing
TreeMetadata object returned by PyTreeCheckpointHandler.metadata. This object mimics and ordinary PyTree to make the change unnoticeable to most users, but also has additional accessible properties not included in any tree mapping operations.
#1276
opened Oct 28, 2024 by
copybara-service
bot
Loading…
Removes the TF dependency on the JaxModule and pushes TF dependencies down to the tensorflow_export. The serving_config processing now also occurs in the tensorflow_export class as it is strictly specific to TensorFlow. TF is still needed in the ExportManager since the public interface allows properties for getting the internal TF module and serving signatures.
#1227
opened Oct 4, 2024 by
copybara-service
bot
Loading…
Introduce
ListKey and TupleKey for list and tuple PyTree nodes respectively.
#1220
opened Oct 1, 2024 by
copybara-service
bot
Loading…
Bump ipython from 7.23.1 to 8.10.0 in /docs/requirements in the pip group across 1 directory
dependencies
Pull requests that update a dependency file
#1201
opened Sep 24, 2024 by
dependabot
bot
Loading…
Add emergency checkpoint logging of arrays for debugging.
#1198
opened Sep 23, 2024 by
copybara-service
bot
Loading…
test restoring to global_mesh in parallel
#1195
opened Sep 19, 2024 by
copybara-service
bot
Loading…
Previous Next
ProTip!
Type g p on any issue or pull request to go back to the pull request listing page.