Commits

Branch selector

 3313c63

User selector

All users

Datepicker

All time

Commit History

Commits on Oct 22, 2024

• No public description

  Show description for 3313c63
  

  liangyaning33

  authored and

  Orbax Authors

  committed
  3313c63

  Copy full SHA for 3313c63

• Improve handlers/args documentation. Also adjust some public symbols to avoid duplicate imports (and resulting breakage) and to get docs to render better.

  Show description for a49a3a0
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  a49a3a0

  Copy full SHA for a49a3a0

Commits on Oct 21, 2024

• Move logging to _src.

  Show description for 71774cd
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  71774cd

  Copy full SHA for 71774cd

• Add announcement for strict option.

  Show description for 7e77f70
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  7e77f70

  Copy full SHA for 7e77f70

Commits on Oct 18, 2024

• Rollback of strict option.

  Show description for ed8f4df
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  ed8f4df

  Copy full SHA for ed8f4df

• Fix slice selection logic in emergency checkpointing, so that a slice with a complete local checkpoint (if it exists) is always chosen as the secondary slice. Additionally, ensure that the `ArrayHa…

  Show description for 559f8a5
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  559f8a5

  Copy full SHA for 559f8a5

• Add strict option in ArrayRestoreArgs, defaulting to True. This prevents arrays from being accidentally padded or truncated when restoring.

  Show description for 77058c3
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  77058c3

  Copy full SHA for 77058c3

• Move multihost package to _src/multihost. Important symbols are still exported publicly.

  Show description for 178a654
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  178a654

  Copy full SHA for 178a654

• Small change to fix test breakage

  Show description for 8cf8779
  

  Orbax Authors

  committed
  8cf8779

  Copy full SHA for 8cf8779

• Adds the ability to specify a checkpoint path to the Orbax export path.

  Show description for 753209e
  

  Orbax Authors

  committed
  753209e

  Copy full SHA for 753209e

Commits on Oct 17, 2024

• Add should_save_fn for emergency CheckpointManager.

  Show description for 1bfb78a
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  1bfb78a

  Copy full SHA for 1bfb78a

• Begins to wire up the new Orbax Model export path to allow exporting using the new Orbax Model pathway.

  Show description for 06b9bd0
  

  Orbax Authors

  committed
  06b9bd0

  Copy full SHA for 06b9bd0

Commits on Oct 16, 2024

• Internal change.

  Show description for 67082f3
  

  niketkumar

  authored and

  Orbax Authors

  committed
  67082f3

  Copy full SHA for 67082f3

• Internal change.

  Show description for 71344e4
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  71344e4

  Copy full SHA for 71344e4

• Internal change.

  Show description for 8ef917e
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  8ef917e

  Copy full SHA for 8ef917e

Commits on Oct 15, 2024

• Use Fragments for serialization.

  Show description for 970e4ff
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  970e4ff

  Copy full SHA for 970e4ff

• Set AsyncOptions.timeout_secs default value to 10 minutes.

  Show description for cb2abb7
  

  niketkumar

  authored and

  Orbax Authors

  committed
  cb2abb7

  Copy full SHA for cb2abb7

• GC potentially left over steps at startup time for local_ckpt_mgr.

  Show description for 71ca8a9
  

  leiyiz

  authored and

  Orbax Authors

  committed
  71ca8a9

  Copy full SHA for 71ca8a9

Commits on Oct 14, 2024

• Reinstate subchunking logic to work around 0-sized dimensions.

  Show description for f4ea9c0
  

  dicentra13

  authored and

  Orbax Authors

  committed
  f4ea9c0

  Copy full SHA for f4ea9c0

Commits on Oct 11, 2024

• Modify enable_pinned_host_transfer option to be provided once for the entire pytree, since it's not really reasonable to customize this on a per-array level.

  Show description for 8b4e90d
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  8b4e90d

  Copy full SHA for 8b4e90d

• Rename _single_item to _default_item in CheckpointManager.

  Show description for fef8e18
  

  BlaziusMaximus

  authored and

  Orbax Authors

  committed
  fef8e18

  Copy full SHA for fef8e18

Commits on Oct 10, 2024

• Add a SaveArgs option that allows disabling pinned host transfer on a per-array basis.

  Show description for 7ffdc88
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  7ffdc88

  Copy full SHA for 7ffdc88

• #cleanup Upgrade type annotations in tensorstore_utils to 3.10.

  Show description for 40ae6ca
  

  dicentra13

  authored and

  Orbax Authors

  committed
  40ae6ca

  Copy full SHA for 40ae6ca

• Refactor build_array_tspec_for_write into a dataclass.

  Show description for 83a2c7c
  

  dicentra13

  authored and

  Orbax Authors

  committed
  83a2c7c

  Copy full SHA for 83a2c7c

Commits on Oct 8, 2024

• Add monitoring to record how long directory creation takes (per sequential instance).

  Show description for 4bb4362
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  4bb4362

  Copy full SHA for 4bb4362

• Refactor calls to choose_chunk_shape to simplify build_zarr_shard_and_chunk_metadata.

  Show description for 4c305e6
  

  dicentra13

  authored and

  Orbax Authors

  committed
  4c305e6

  Copy full SHA for 4c305e6

Commits on Oct 7, 2024

• Now that JAX supports donate on device_put, use that instead of the ugly device_put -> block -> delete routine

  Show description for db46fc7
  

  yashk2810

  authored and

  Orbax Authors

  committed
  db46fc7

  Copy full SHA for db46fc7

• Remove orbax-checkpoint support for Python 3.9.

  Show description for d1b8aaa
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  d1b8aaa

  Copy full SHA for d1b8aaa

Commits on Oct 3, 2024

• Refine several API docs and add a test to prevent colab docs from breaking.

  Show description for 410ca56
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  410ca56

  Copy full SHA for 410ca56

• Fix _determine_single_item_mode_from_directory by incorporating step name format.

  Show description for 8bf99e3
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  8bf99e3

  Copy full SHA for 8bf99e3

Commits on Oct 2, 2024

• Rename CheckpointMetadata to StepMetadata.

  Show description for 367ad4d
  

  BlaziusMaximus

  authored and

  Orbax Authors

  committed
  367ad4d

  Copy full SHA for 367ad4d

• Allow registering an item in DefaultCheckpointHandlerRegistry without providing the actual handler, as long as the provided args correspond to a globally registered handler. This allows for sligh…

  Show description for 07b17c1
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  07b17c1

  Copy full SHA for 07b17c1

• Support empty tuple values in checkpointable params.

  Show description for aad5e72
  

  niketkumar

  authored and

  Orbax Authors

  committed
  aad5e72

  Copy full SHA for aad5e72

• Use CheckpointHandlerRegistry when CheckpointManager is default-constructed, e.g. CheckpointManager(directory). Previously this usage pattern would default to single-item mode, but the new be…

  Show description for a01eab8
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  a01eab8

  Copy full SHA for a01eab8

• No public description

  Show description for e40ba06
  

  Orbax Authors

  committed
  e40ba06

  Copy full SHA for e40ba06

Pagination

Previous

Next