-
Notifications
You must be signed in to change notification settings - Fork 28
Pull requests: google/orbax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
test restoring to global_mesh in parallel
#1195
opened Sep 19, 2024 by
copybara-service
bot
Loading…
Add support for
shard_axes
into build_array_tspec_for_write
, to allow subchunking of arrays on specific axes when writing to storage.
#1194
opened Sep 19, 2024 by
copybara-service
bot
Loading…
Remove logging of ts_spec in validation step - this is very verbose.
#1191
opened Sep 18, 2024 by
copybara-service
bot
Loading…
Use
CheckpointHandlerRegistry
when CheckpointManager
is default-constructed, e.g. CheckpointManager(directory)
. Previously this usage pattern would default to single-item mode, but the new behavior will default to lazy initialization. The change should be a no-op for existing users.
#1177
opened Sep 13, 2024 by
copybara-service
bot
Loading…
Use thread-safe barrier api by default in Orbax.
#1175
opened Sep 13, 2024 by
copybara-service
bot
Loading…
Add an option to skip validating merged parameters in Orbax.
#1126
opened Sep 2, 2024 by
copybara-service
bot
Loading…
Include directory path in
create_directory
barrier key.
#1040
opened Aug 5, 2024 by
copybara-service
bot
Loading…
Modernize typing in composite checkpoint handler.
#1014
opened Jul 22, 2024 by
copybara-service
bot
Loading…
Previous Next
ProTip!
no:milestone will show everything without a milestone.