1 Release published by 1 person

• v0.11.13

  published May 1, 2025

15 Pull requests merged by 1 person

• Fix SaveDecisionPolicy docstrings for readthedocs.

  #1891 merged May 2, 2025

• Add additional batching components in obm export

  #1886 merged May 1, 2025

• #v1 Fix test handlers that were writing a file on all processes.

  #1888 merged May 1, 2025

• Increase Orbax version.

  #1885 merged May 1, 2025

• Change several warning messages to info messages, as these are surfaced far too often and give users the mistaken impression of doing something wrong.

  #1884 merged May 1, 2025

• #v1 Rename GenericHandler to StatefulCheckpointableHandler, remove metadata method, and other minor changes.

  #1883 merged Apr 30, 2025

• Replace GlobalSupplemental & GlobalSupplementalMap with map<string, UnstructuredData> in manifest.

  #1857 merged Apr 30, 2025

• #v1 Add StatefulCheckpointable interface and GenericHandler, and implement save & load methods within GenericHandler.

  #1861 merged Apr 29, 2025

• Fix code snippet rendering in a couple of docstrings.

  #1869 merged Apr 29, 2025

• #v1 Enable v1 users to load v0 checkpoints without using contexts.

  #1879 merged Apr 29, 2025

• Update JAX nightly index usage

  #1878 merged Apr 29, 2025

• No public description

  #1874 merged Apr 28, 2025

• #v1 Add compatibility tests for save-by-v0-load-by-v1 and also fix code.

  #1849 merged Apr 28, 2025

• #v1 Add checkpointables support for training.Checkpointer.

  #1854 merged Apr 28, 2025

• Add num_batch_threads and max_enqueued_batches to OBM batching configs.

  #1868 merged Apr 28, 2025

9 Pull requests opened by 1 person

• Add new float8e3m4 dtype for testing

  #1873 opened Apr 28, 2025

• #v1 Add checkpointables support for `training.Checkpointer`.

  #1875 opened Apr 28, 2025

• Re-organize OCP v0 and v1 docs directory structure.

  #1876 opened Apr 28, 2025

• Use the new preservation policy in CheckpointManager.

  #1880 opened Apr 29, 2025

• #v1 Partially port the existing checkpoint format page into V1 documentation, but most of it is rewritten and updated for V1.

  #1881 opened Apr 29, 2025

• Create API reference docs for orbax.checkpoint.experimental.v1

  #1887 opened May 1, 2025

• Enforce that the input and output name are not optional and remove the logic to deal with non input name and output name logic.

  #1889 opened May 1, 2025

• #v1 Allow `load_pytree` to specify the checkpointable_name, with `pytree` as default value.

  #1890 opened May 2, 2025

• Fix v1 options docstrings for readthedocs.

  #1892 opened May 2, 2025

1 Issue closed by 1 person

• Optimistic Adam optimizer broken when using scheduled learning rate

  #1882 closed Apr 30, 2025

1 Unresolved conversation

Sometimes conversations happen on old items that aren’t yet closed. Here is a list of all the Issues and Pull Requests with unresolved conversations.

• Updated save and restore signatures for Emergency Checkpoint Manager

  #1852 commented on Apr 29, 2025 • 0 new comments