Commits

Branch selector

 7a400d1

User selector

All users

Datepicker

All time

Commit History

Commits on Nov 27, 2024

• No public description

  Show description for 7a400d1
  

  ChromeHearts

  authored and

  Orbax Authors

  committed
  7a400d1

• Minor fixes to metadata serialization.

  Show description for 6a66183
  

  BlaziusMaximus

  authored and

  Orbax Authors

  committed
  6a66183

Commits on Nov 26, 2024

• Internal

  Show description for 8a34b74
  

  liangyaning33

  authored and

  Orbax Authors

  committed
  8a34b74

• No public description

  Show description for 07a7303
  

  niketkumar

  authored and

  Orbax Authors

  committed
  07a7303

• No public description

  Show description for 7c67abf
  

  liangyaning33

  authored and

  Orbax Authors

  committed
  7c67abf

• No public description

  Show description for 0ae565d
  

  maxwillzq

  authored and

  Orbax Authors

  committed
  0ae565d

Commits on Nov 25, 2024

• Internal change.

  Show description for 93f1b91
  

  niketkumar

  authored and

  Orbax Authors

  committed
  93f1b91

• Fix class name of a namedtuple created on the fly by replacing . with _.

  Show description for 56e11cf
  

  niketkumar

  authored and

  Orbax Authors

  committed
  56e11cf

• Reflect #1358, which enables replica_parallel saving. Uses cache to avoid recomputing devices_indices_map for arrays with identical shape/sharding.

  Show description for 46d159e
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  46d159e

• Fix namedtuple empty value typestr when experimental support_rich_types is disabled again after enabling it.

  Show description for 3a32a23
  

  niketkumar

  authored and

  Orbax Authors

  committed
  3a32a23

Commits on Nov 24, 2024

• No public description

  Show description for e4a1239
  

  ChromeHearts

  authored and

  Orbax Authors

  committed
  e4a1239

Commits on Nov 23, 2024

• [Experimental Feature] Support empty NamedTuple leaf when PyTreeMetadataOptions.support_rich_types=true.

  Show description for ab94736
  

  niketkumar

  authored and

  Orbax Authors

  committed
  ab94736

Commits on Nov 22, 2024

• fix BUILD files.

  Show description for b3d29c6
  

  liangyaning33

  authored and

  Orbax Authors

  committed
  b3d29c6

• Add docstring for CheckpointManagerOptions.temporary_path_class.

  Show description for ca9b7a6
  

  Orbax Authors

  committed
  ca9b7a6

• Add /checkpoint/_src/path/snapshot/BUILD and /checkpoint/_src/path/BUILD.

  Show description for 309c59f
  

  liangyaning33

  authored and

  Orbax Authors

  committed
  309c59f

• No public description

  Show description for 5048d35
  

  liangyaning33

  authored and

  Orbax Authors

  committed
  5048d35

• Add tests and documentation for abstract_arrays.

  Show description for 1704519
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  1704519

• ServingConfig.extra_trackable_resources is now supported in the Orbax Model path when support_tf_resources is True.

  Show description for 1cb2786
  

  wangpengmit

  authored and

  Orbax Authors

  committed
  1cb2786

• Internal change.

  Show description for 922d408
  

  Orbax Authors

  committed
  922d408

• Release 0.10.1.

  Show description for 65c9068
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  65c9068

• Increase version to 0.10.1 to incorporate emergency checkpointing fix.

  Show description for 61c66fd
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  61c66fd

• Add an option to disable saving root-level metadata in CheckpointManager.

  Show description for efbb7b2
  

  BlaziusMaximus

  authored and

  Orbax Authors

  committed
  efbb7b2

• Correct issue where emergency checkpoint debug logging was not OSS-ed.

  Show description for aaa93e9
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  aaa93e9

• [emergency checkpoint] Add ReplicatorCheckpointManager implementation for interoperating with replicator service provided by GKE (or theoretically, any other similar service).

  Show description for bf6f388
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  bf6f388

• Add shape property to ArrayRestoreArgs to be interchangeable with global_shape. This allows generic functions like to_shape_dtype_struct to reference it.

  Show description for 8055aaf
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  8055aaf

• Add support to Layout using StandardCheckpointHandler

  Show description for fa00f61
  

  ChromeHearts

  authored and

  Orbax Authors

  committed
  fa00f61

• Release 0.10.0 and correct the changelog.

  Show description for bdb77e8
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  bdb77e8

• Add logging to detect missing chunks to emergency checkpointing to facilitate local checkpoint debugging.

  Show description for deb0b73
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  deb0b73

• Add validation to prevent loading an array index that was never written to.

  Show description for 646f1e6
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  646f1e6

Commits on Nov 21, 2024

• setup github actions to periodicly run existing OSS tests every hour.

  Show description for 04e4ae7
  

  liangyaning33

  authored and

  Orbax Authors

  committed
  04e4ae7

• Expose public symbols via orbax/checkpoint/type_handlers.py

  Show description for 8093d4d
  

  ChromeHearts

  authored and

  Orbax Authors

  committed
  8093d4d

• Change XlaRuntimeError to jax.errors.JaxRuntimeError

  Show description for 740af7a
  

  Jake VanderPlas

  authored and

  Orbax Authors

  committed
  740af7a

• Merge #1320 which adds replica-parallel saving for arrays with replicated shards. Thanks to https://github.com/gspschmid.

  Show description for 477a683
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  477a683

• Merge pull request #1320 from gspschmid/gschmid/replica_parallel_1

  Show description for 906ec19
  cpgaffney1

  authored
  906ec19

• Add vlogs for emergency checkpointing so we can easily tell which indices of the array need to be loaded.

  Show description for a4449d6
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  a4449d6

Pagination

Previous

Next