Commits

Branch selector

 a4449d6

User selector

All users

Datepicker

All time

Commit History

Commits on Nov 21, 2024

• 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

• Add .bazelrc, .bazelversion, requirements.txt, WORKSPACE to orbax github repo.

  Show description for 15f18e7
  

  liangyaning33

  authored and

  Orbax Authors

  committed
  15f18e7

Commits on Nov 20, 2024

• Refactor and restructure constructs from type_handlers.py and metadata

  Show description for 2a39f3a
  

  niketkumar

  authored and

  Orbax Authors

  committed
  2a39f3a

• Adds a new support_tf_resources flag (defaults to False) to jax2obm_kwargs to enable a new way to save TF (concrete) functions in Orbax Model that allows those functions to use TF resources (…

  Show description for 71b1ba0
  

  wangpengmit

  authored and

  Orbax Authors

  committed
  71b1ba0

• [Experimental Feature] Support NamedTuple and Tuple nodes in PyTree metadata.

  Show description for 4a3fca2
  

  niketkumar

  authored and

  Orbax Authors

  committed
  4a3fca2

Commits on Nov 19, 2024

• Move Checkpointer implementations to _src.

  Show description for 8c4d71b
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  8c4d71b

• Internal

  Show description for 9c5deb3
  

  Orbax Authors

  committed
  9c5deb3

• Add/Update tests for is_empty_or_leaf and is_empty_node.

  Show description for 433743a
  

  niketkumar

  authored and

  Orbax Authors

  committed
  433743a

Commits on Nov 18, 2024

• Create a separate namespace package orbax.checkpoint.testing for exporting test objects.

  Show description for acb6607
  

  niketkumar

  authored and

  Orbax Authors

  committed
  acb6607

• Internal change.

  Show description for 790d0e2
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  790d0e2

• Add support for legacy metadata file path in CheckpointManager.

  Show description for 86a74b8
  

  BlaziusMaximus

  authored and

  Orbax Authors

  committed
  86a74b8

• Wrap tree.utils.serialize_tree -> metadata.tree.serialize_tree to allow NamedTuple and Tuple node types in PyTree metadata.

  Show description for 0e737f0
  

  niketkumar

  authored and

  Orbax Authors

  committed
  0e737f0

• Internal change.

  Show description for ffa7e69
  

  niketkumar

  authored and

  Orbax Authors

  committed
  ffa7e69

Commits on Nov 15, 2024

• No public description

  Show description for 0f9e562
  

  Orbax Authors

  committed
  0f9e562

• Add chex as an optional testing dependency to unbreak RTD. Unclear why this has not previously been a problem.

  Show description for 8c02199
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  8c02199

• No public description

  Show description for 9aa1e58
  

  niketkumar

  authored and

  Orbax Authors

  committed
  9aa1e58

• Add Layout support

  Show description for 863452b
  

  ChromeHearts

  authored and

  Orbax Authors

  committed
  863452b

• Updated Version

  Show description for 577ebf2
  

  Orbax Authors

  committed
  577ebf2

• No public description

  Show description for db18acb
  

  Orbax Authors

  committed
  db18acb

Commits on Nov 14, 2024

• Internal reflection of #1319, with a few adjustments. All credit and thanks to https://github.com/gspschmid.

  Show description for e257b2d
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  e257b2d

• Merge pull request #1319 from gspschmid/gschmid/replica_parallel_0

  Show description for acd7869
  cpgaffney1

  authored
  acd7869

• Remove locking and its dependencies.

  Show description for 33d9bb4
  

  liangyaning33

  authored and

  Orbax Authors

  committed
  33d9bb4

• Add user-facing Root/StepMetadata classes and internal serialization modules.

  Show description for 7ffa18f
  

  BlaziusMaximus

  authored and

  Orbax Authors

  committed
  7ffa18f

• Break out mesh construction and process ID metadata utils into a separate file.

  Show description for 0a62ed8
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  0a62ed8

• Refactor metadata/tree_test.py and move common test types to test_tree_utils.py for better reusability.

  Show description for c65c761
  

  niketkumar

  authored and

  Orbax Authors

  committed
  c65c761

Commits on Nov 13, 2024

• Fix device to HBM mapping for Trillium.

  Show description for 58d423f
  

  abhinavclemson

  authored and

  Orbax Authors

  committed
  58d423f

• Fix emergency checkpointing issue arising when repeatedly restoring from a local checkpoint. Process ID remapping may happen in different ways on subsequent times, so we need to ensure new process …

  Show description for 7f2cf3c
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  7f2cf3c

• [replica-parallel] Add replica slices concept

  gspschmid

  committed
  5f533d5

• Add a distinct shape/dtype struct type for Numpy arrays.

  Show description for 0e54f53
  

  dicentra13

  authored and

  Orbax Authors

  committed
  0e54f53

• No public description

  Show description for 733f0d4
  

  Orbax Authors

  committed
  733f0d4

Commits on Nov 12, 2024

• Internal change.

  Show description for 19f53a2
  

  niketkumar

  authored and

  Orbax Authors

  committed
  19f53a2

• Manual submit of #1323.

  Show description for a160eb8
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  a160eb8

• Internal change.

  Show description for 174d6aa
  

  niketkumar

  authored and

  Orbax Authors

  committed
  174d6aa

Commits on Nov 11, 2024

• Fix readthedoc build issue

  Show description for c2d6863
  

  ChromeHearts

  authored and

  Orbax Authors

  committed
  c2d6863

• Refactor *pytree*handler.py to improve readability of metadata/tree api usage.

  Show description for 9e18554
  

  niketkumar

  authored and

  Orbax Authors

  committed
  9e18554

Pagination

Previous

Next