Pull requests: google/orbax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Allow
AsyncCheckpointer
to work correctly when there are multiple instances. Previously a deadlock could be observed between write/commit barriers.
#900
opened May 24, 2024 by
copybara-service
bot
Loading…
Support non blocking CheckpointMetadataStore.write.
#899
opened May 24, 2024 by
copybara-service
bot
Loading…
End support for legacy Pax formats with directory structure:
#882
opened May 16, 2024 by
copybara-service
bot
Loading…
Fork a small amount of Orbax code into Pax to deal with writing "aggregate" files, as Orbax will soon lose this ability.
#869
opened May 8, 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…
Add OCDBT support to GlobalHostArray type_handler
#843
opened Apr 24, 2024 by
copybara-service
bot
Loading…
Create the utils function load_model_with_tfrt API for XLA devices.
#813
opened Apr 12, 2024 by
copybara-service
bot
Loading…
Enable Orbax checkpointing with background deleting thread
#796
opened Apr 4, 2024 by
copybara-service
bot
Loading…
Create an interface for retrieving the current step number.
#778
opened Mar 30, 2024 by
copybara-service
bot
Loading…
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…
Fix errors of loading ckpt that contains sharding configs.
#671
opened Jan 17, 2024 by
copybara-service
bot
Loading…
Replace references to deprecated device_buffer attributes
#616
opened Dec 6, 2023 by
copybara-service
bot
Loading…
Internal changes to support more export / load options.
#593
opened Nov 17, 2023 by
copybara-service
bot
Loading…
Previous Next
ProTip!
Exclude everything labeled
bug
with -label:bug.