8000 Commits · google/orbax · GitHub

More Web Proxy on the site http://driver.im/

Commits

Commits on May 13, 2025

#v1 add test cases for `is_handable` for Checkpointable interface.

liangyaning33
authored and
Orbax Authors
committed
Generalize supplemental info in the StableHLO body. This is to provide the flexibility to store more than one type of supplemental info in the future.

qdhack
authored and
Orbax Authors
committed

Commits on May 12, 2025

clarify docstring of finalize impls
Orbax Authors
committed
#v1 Update `training` guide to reference the Grain iterator.

cpgaffney1
authored and
Orbax Authors
committed
Reduce warning log to vlog in key_value_try_get

mridul-sahu
authored and
Orbax Authors
committed
Internal change.

cpgaffney1
authored and
Orbax Authors
committed

Commits on May 9, 2025

Use `.update()` on `jax.ShapeDtypeStruct` instead of assigning to it directly.

yashk2810
authored and
Orbax Authors
committed
No public description

Nikitas Rontsis
authored and
Orbax Authors
committed
Enforce that the input and output name are not optional and remove the logic to deal with non input name and output name logic.
Orbax Authors
committed

Commits on May 8, 2025

Internal Change

ChromeHearts
authored and
Orbax Authors
committed
Add PLACEHOLDER to v1 api to enable partial restoration.

BlaziusMaximus
authored and
Orbax Authors
committed
Add v1 ArrayLeafHandler

ChromeHearts
authored and
Orbax Authors
committed
#v1 Port async checkpointing doc from v0 to v1.

niketkumar
authored and
Orbax Authors
committed
[NFC] Add `# pytype: disable` annotations because of an upcoming change where PartitionSpec no longer inherits from a tuple. This is already true at runtime. No functional changes intended!

yashk2810
authored and
Orbax Authors
committed
Copy full SHA for 3e4d334
Propagate JaxRuntimeError to callers in SignalingClient.

mridul-sahu
authored and
Orbax Authors
committed
Add a manifest_version.txt file in the saved directory.
Orbax Authors
committed

Commits on May 7, 2025

#v1 Add `start_async_mkdir` that begins async directory creation and returns a `PathAwaitingCreation`.

cpgaffney1
authored and
Orbax Authors
committed
#v1 Update the checkpointables documentation to include a Grain iterator example.

cpgaffney1
authored and
Orbax Authors
committed
Modify the definition of BatchComponent.WHOLE_PIPELINE.
Orbax Authors
committed
Fix Random Key restoration with StandardRestore

ChromeHearts
authored and
Orbax Authors
committed
Add PreservationPolicy to v1, use PolicyCheckpointInfo instead of CheckpointInfo.

BlaziusMaximus
authored and
Orbax Authors
committed
#v1 Add guide for `ocp.training` package.

cpgaffney1
authored and
Orbax Authors
committed

Commits on May 6, 2025

[orbax] replace uses of deprecated jax.util APIs

Jake VanderPlas
authored and
Orbax Authors
committed
No public description

tayoola
authored and
Orbax Authors
committed
#v1 Allow `load_pytree` to specify the checkpointable_name, with `pytree` as default value.

niketkumar
authored and
Orbax Authors
committed

Commits on May 5, 2025

Remove the FunctionDef-based approach in tf2obm
Orbax Authors
committed
#v1 Add `synchronize_next_operation_id`.

cpgaffney1
authored and
Orbax Authors
committed
#v1 Add metadata async write / read methods.

cpgaffney1
authored and
Orbax Authors
committed
Remove World Readable bit checkpoints instead of letting model directories inherit World Readable bits from parent folder.
Orbax Authors
committed
#v1 Partially port the existing checkpoint format page into V1 documentation, but most of it is rewritten and updated for V1.

cpgaffney1
authored and
Orbax Authors
committed
Create API reference docs for orbax.checkpoint.experimental.v1

BlaziusMaximus
authored and
Orbax Authors
committed
Fix v1 options docstrings for readthedocs.

BlaziusMaximus
authored and
Orbax Authors
committed

Commits on May 3, 2025

Some internal changes.
Orbax Authors
committed

Commits on May 2, 2025

Fix SaveDecisionPolicy docstrings for readthedocs.

BlaziusMaximus
authored and
Orbax Authors
committed

Commits on May 1, 2025

Add additional batching components in obm export
Orbax Authors
committed

0