-
Notifications
You must be signed in to change notification settings - Fork 44
Pull requests: google/orbax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
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…
[orbax model] Support multiple python and jax versions on Github Actions.
#1671
opened Mar 6, 2025 by
copybara-service
bot
Loading…
Make
custom_metadata field naming consistent.
#1670
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…
Remove descriptor creation from CompositeCheckpointHandler
#1665
opened Mar 6, 2025 by
copybara-service
bot
Loading…
Fix ReadTheDocs breakage and fix a number of formatting issues. Refactor
orbax.checkpoint.logging in service of this objective.
#1664
opened Mar 6, 2025 by
copybara-service
bot
Loading…
Simplify heartbeat options for PjRt distributed runtime.
#1663
opened Mar 5, 2025 by
copybara-service
bot
Loading…
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…
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…
Support minimum_interval_seconds since last save in Continuous checkpoint save policy.
#1645
opened Mar 3, 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…
Add save barrier to ensure checkpoint manager and checkpointers save start is synchronized on all processes.
#1629
opened Feb 25, 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…
Previous Next
ProTip!
Updated in the last three days: updated:>2025-03-03.