Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Jul 15, 2023

• Internal change

  Show description for 4e3a99f
  

  yashk2810

  authored and

  copybara-github

  committed
  4e3a99f

  Copy full SHA for 4e3a99f

Commits on Jul 14, 2023

• Move TensorSpecWithDefault and its util functions to utils.py, as this is generally useful and can be used without other Orbax classes like JaxModule.

  Show description for 5ad6cba
  

  voutcn

  authored and

  copybara-github

  committed
  5ad6cba

  Copy full SHA for 5ad6cba

• Fix bug in _wait_for_new_checkpoint.

  Show description for 769d4d7
  

  Orbax Authors

  authored and

  copybara-github

  committed
  769d4d7

  Copy full SHA for 769d4d7

Commits on Jul 13, 2023

• Identify only needed input parameters to restore during restoration with transformations. This was already being done, but explicitly identifying the needed parameters will be helpful for upcoming …

  Show description for a6484ec
  

  cpgaffney1

  authored and

  copybara-github

  committed
  a6484ec

  Copy full SHA for a6484ec

• Internal change

  Show description for dd6348e
  

  faizan-m

  authored and

  copybara-github

  committed
  dd6348e

  Copy full SHA for dd6348e

Commits on Jul 12, 2023

• Use `import orbax.checkpoint as ocp` to avoid the verbosity of using 'orbax.checkpoint` every time.

  Show description for bc8f8b4
  

  cpgaffney1

  authored and

  copybara-github

  committed
  bc8f8b4

  Copy full SHA for bc8f8b4

Commits on Jul 11, 2023

• Fix linkcode resolution, previously specific to checkpoint only.

  Show description for 8bb745b
  

  ajain-23

  authored and

  copybara-github

  committed
  8bb745b

  Copy full SHA for 8bb745b

• Split orbax_export doc into OSS and google-internal versions. To make review easy, this is pure document refactoring. We can continue improve the colab notebook later.

  Show description for 3644bcb
  

  maxwillzq

  authored and

  copybara-github

  committed
  3644bcb

  Copy full SHA for 3644bcb

Commits on Jul 10, 2023

• Add OCDBT usage documentation to the RTD page.

  Show description for 1c56e81
  

  cpgaffney1

  authored and

  copybara-github

  committed
  1c56e81

  Copy full SHA for 1c56e81

• Add `close` method for Checkpointer and CheckpointHandler. Add context manager helper functions for Checkpointer and CheckpointManager.

  Show description for 670c7dd
  

  cpgaffney1

  authored and

  copybara-github

  committed
  670c7dd

  Copy full SHA for 670c7dd

Commits on Jun 30, 2023

• Add table of contents for readthedocs page and autogenerate logo to make for easier navigation.

  Show description for c7a7fd4
  

  cpgaffney1

  authored and

  copybara-github

  committed
  c7a7fd4

  Copy full SHA for c7a7fd4

• Remove duplicate installation guide on `index.rst` and point users to README.md.

  Show description for 97a4810
  

  maxwillzq

  authored and

  copybara-github

  committed
  97a4810

  Copy full SHA for 97a4810

Commits on Jun 29, 2023

• Speed up CheckpointManager creation for directories with many checkpoints.

  Show description for d480056
  

  Orbax Authors

  authored and

  copybara-github

  committed
  d480056

  Copy full SHA for d480056

• Re-enable the dtensor export unit test with native serialization.

  Show description for 4873705
  

  voutcn

  authored and

  copybara-github

  committed
  4873705

  Copy full SHA for 4873705

Commits on Jun 28, 2023

• Expose TensorSpecWithDefault to Orbax Export's __init__.

  Show description for 5db49eb
  

  Orbax Authors

  authored and

  copybara-github

  committed
  5db49eb

  Copy full SHA for 5db49eb

• Increase orbax.checkpoint version.

  Show description for afc4b61
  

  cpgaffney1

  authored and

  copybara-github

  committed
  afc4b61

  Copy full SHA for afc4b61

• Set default types in Flax for Orbax restoration and add `restore_with_serialized_types` in preparation for an upcoming change.

  Show description for e195d98
  

  cpgaffney1

  authored and

  copybara-github

  committed
  e195d98

  Copy full SHA for e195d98

• Open-source ProtoCheckpointHandler.

  Show description for 054df82
  

  cpgaffney1

  authored and

  copybara-github

  committed
  054df82

  Copy full SHA for 054df82

• Also check DTensor shard shapes in dtensor_utils unit test.

  Show description for c90b4d0
  

  voutcn

  authored and

  copybara-github

  committed
  c90b4d0

  Copy full SHA for c90b4d0

Commits on Jun 27, 2023

• Relax JaxArray -> DTensor conversion requirement: allow sharding one tensor dimension to multiple mesh axes if there is at most one axis with size > 1.

  Show description for fb888ef
  

  voutcn

  authored and

  copybara-github

  committed
  fb888ef

  Copy full SHA for fb888ef

• Add api_reference for orbax_epxort.

  Show description for 5b49be8
  

  maxwillzq

  authored and

  copybara-github

  committed
  5b49be8

  Copy full SHA for 5b49be8

• Fix unlocking function, which may fail if there are multiple evaluators

  Show description for ee60bcd
  

  cpgaffney1

  authored and

  copybara-github

  committed
  ee60bcd

  Copy full SHA for ee60bcd

• If there is only one serving config, attach the serving function to the module to be exported, so that the model can be loaded as a callable python module.

  Show description for b2e8ede
  

  voutcn

  authored and

  copybara-github

  committed
  b2e8ede

  Copy full SHA for b2e8ede

Commits on Jun 26, 2023

• Eliminates hosts sync on background thread and fixes issue with reading lockfile before checkpoint is finalized.

  Show description for 4726dcc
  

  cpgaffney1

  authored and

  copybara-github

  committed
  4726dcc

  Copy full SHA for 4726dcc

• Allow value_fn and multi_value_fn to accept RestoreArgs as an argument when they are used during restore, so that the user may customize the returned value based on what is requested by RestoreArgs.

  Show description for b4a9c37
  

  cpgaffney1

  authored and

  copybara-github

  committed
  b4a9c37

  Copy full SHA for b4a9c37

Commits on Jun 23, 2023

• Internal change

  Show description for 722f416
  

  cpgaffney1

  authored and

  copybara-github

  committed
  722f416

  Copy full SHA for 722f416

• Internal change.

  Show description for ab16766
  

  cpgaffney1

  authored and

  copybara-github

  committed
  ab16766

  Copy full SHA for ab16766

• Update READMEs.

  Show description for c7df7ca
  

  ajain-23

  authored and

  copybara-github

  committed
  c7df7ca

  Copy full SHA for c7df7ca

Commits on Jun 22, 2023

• https://github.com/google/orbax/issues/364: Move TF to optional dependencies.

  Show description for 7b826cb
  

  maxwillzq

  authored and

  copybara-github

  committed
  7b826cb

  Copy full SHA for 7b826cb

• Alter behavior of Orbax `wait_for_new_checkpoint`. Use Orbax checkpoint_utils methods in Pax eval/decode.

  Show description for ea3aab8
  

  cpgaffney1

  authored and

  copybara-github

  committed
  ea3aab8

  Copy full SHA for ea3aab8

• Fix issue with `multi_value_fn` in restoration where an input value could be loaded many times unnecessarily.

  Show description for ca528b0
  

  cpgaffney1

  authored and

  copybara-github

  committed
  ca528b0

  Copy full SHA for ca528b0

• Clean up checkpointing dependencies.

  Show description for 137d537
  

  cpgaffney1

  authored and

  copybara-github

  committed
  137d537

  Copy full SHA for 137d537

Commits on Jun 20, 2023

• Remove unused dependencies.

  Show description for ab34949
  

  cpgaffney1

  authored and

  copybara-github

  committed
  ab34949

  Copy full SHA for ab34949

Commits on Jun 16, 2023

• Release.

  Show description for 311c7c8
  

  cpgaffney1

  authored and

  copybara-github

  committed
  311c7c8

  Copy full SHA for 311c7c8

• Support creating sharded array when ArrayRestoreArgs is passed and the value

  Show description for 64da135
  

  cpgaffney1

  authored and

  copybara-github

  committed
  64da135

  Copy full SHA for 64da135

Pagination

Previous

Next