Thanks to visit codestin.com
Credit goes to github.com

Commits

Commits on Jun 14, 2024

Use only one directory creation request per item rather than 1 per item per host.

cpgaffney1
authored and
Orbax Authors
committed

Commits on Jun 13, 2024

Internal change.

cpgaffney1
authored and
Orbax Authors
committed
For `broadcast_one_to_all`, set default `is_source = process_index() == 0` to avoid discrepancies between process index used by Orbax and process index used by underlying JAX `broadcast_one_to_all`.

cpgaffney1
authored and
Orbax Authors
committed

Commits on Jun 12, 2024

Revert: Improve tensorstore I/O efficiency
Orbax Authors
committed
Revert OCDBT manifest_kind change
Orbax Authors
committed
Release new Orbax version.

cpgaffney1
authored and
Orbax Authors
committed
Internal change.

cpgaffney1
authored and
Orbax Authors
committed

Commits on Jun 11, 2024

Improve tensorstore I/O efficiency
Orbax Authors
committed
Internal change.

cpgaffney1
authored and
Orbax Authors
committed
Add checkpoint format docs page.

cpgaffney1
authored and
Orbax Authors
committed

Commits on Jun 10, 2024

Add OBX_EXPORT_TF_PREPROCESS_ONLY Mode to support debugging.

maxwillzq
authored and
Orbax Authors
committed
Swap `jax.process_index()` for `jax._src.distributed.global_state.process_id`.

cpgaffney1
authored and
Orbax Authors
committed

Commits on Jun 7, 2024

No public description

cpgaffney1
authored and
Orbax Authors
committed
Enable standard logger by default and Fix `synchronous` flag input data in logging.

abhinavclemson
authored and
Orbax Authors
committed

Commits on Jun 6, 2024

Refactor functions from `utils` to `tree` package.

cpgaffney1
authored and
Orbax Authors
committed

Commits on Jun 5, 2024

Internal change.

cpgaffney1
authored and
Orbax Authors
committed

Commits on Jun 4, 2024

Track more statistics of the delete duration

ChromeHearts
authored and
Orbax Authors
committed
Fix a typo in exception message.
Orbax Authors
committed
No public description

liangyaning33
authored and
Orbax Authors
committed

Commits on Jun 3, 2024

Internal change.

cpgaffney1
authored and
Orbax Authors
committed

Commits on May 31, 2024

New Orbax release.

cpgaffney1
authored and
Orbax Authors
committed
Internal change.

niketkumar
authored and
Orbax Authors
committed

Commits on May 30, 2024

Adjust test_utils_test to account for internal/external discrepancy.

cpgaffney1
authored and
Orbax Authors
committed

Commits on May 29, 2024

Use `jax.tree.*` instead of `jax.tree_util.tree_*`, as the latter is now deprecated.

cpgaffney1
authored and
Orbax Authors
committed
Internal change.

niketkumar
authored and
Orbax Authors
committed
Internal change.
Orbax Authors
committed
No public description

niketkumar
authored and
Orbax Authors
committed

Commits on May 28, 2024

Internal change.

cpgaffney1
authored and
Orbax Authors
committed
Allow `AsyncCheckpointer` to work correctly when there are multiple instances. Previously a deadlock could be observed between write/commit barriers.

cpgaffney1
authored and
Orbax Authors
committed
No public description

liangyaning33
authored and
Orbax Authors
committed

Commits on May 26, 2024

Fix small issue when sharding is None.

maxwillzq
authored and
Orbax Authors
committed

Commits on May 24, 2024

Introduce `FileOptions` as CheckpointManagerOptions attribute.

niketkumar
authored and
Orbax Authors
committed
Modify _write_metadata_file to async.

liangyaning33
authored and
Orbax Authors
committed

Commits on May 23, 2024

No public description

liangyaning33
authored and
Orbax Authors
committed
Increase orbax version.

cpgaffney1
authored and
Orbax Authors
committed