Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Jan 5, 2025

• No public description

  Show description for 58d7b26
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  58d7b26

Commits on Jan 3, 2025

• In the Orbax Model (OBM) mode of Orbax Export, saves TF SavedModel to a sub-folder instead of the main OBM folder.

  Show description for 3fa494f
  

  wangpengmit

  authored and

  Orbax Authors

  committed
  3fa494f

Commits on Jan 2, 2025

• Add user-facing TreeMetadata object returned by PyTreeCheckpointHandler.metadata. This object mimics an ordinary PyTree to make the change unnoticeable to most users, but also has additional ac…

  Show description for 605185a
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  605185a

• Add tests that don't require real accelerators to allow future reproducers.

  Show description for 002ed43
  

  Orbax Authors

  committed
  002ed43

• No public description

  Show description for c4e1c51
  

  Conchylicultor

  authored and

  Orbax Authors

  committed
  c4e1c51

Commits on Dec 30, 2024

• Resolve default value of Handler.typestr if method missing.

  Show description for 32f2c03
  

  niketkumar

  authored and

  Orbax Authors

  committed
  32f2c03

Commits on Dec 24, 2024

• Change the Tensorflow export function from lambda function to normal function definition.

  Show description for c227788
  

  Orbax Authors

  committed
  c227788

• No public description

  Show description for 27fae17
  

  Nikitas Rontsis

  authored and

  Orbax Authors

  committed
  27fae17

Commits on Dec 19, 2024

• Make Self import compatible with Python < 3.11

  Show description for f85f6ca
  

  ChromeHearts

  authored and

  Orbax Authors

  committed
  f85f6ca

• Improve test coverage of type validation in metadata deserialization.

  Show description for b79a298
  

  BlaziusMaximus

  authored and

  Orbax Authors

  committed
  b79a298

Commits on Dec 18, 2024

• Remove jax_enable_memories flag

  Show description for f7bece0
  

  yashk2810

  authored and

  Orbax Authors

  committed
  f7bece0

• Expand StepMetadata.item_metadata type to include non-composite cases.

  Show description for a2d4160
  

  BlaziusMaximus

  authored and

  Orbax Authors

  committed
  a2d4160

• Expand StepMetadata.item_handlers type to include non-composite cases.

  Show description for 6faf7af
  

  BlaziusMaximus

  authored and

  Orbax Authors

  committed
  6faf7af

Commits on Dec 16, 2024

• [Subchunking] Allow to chunk abstract fragments, and ensure that the order of returned chunks is defined and deterministic.

  Show description for a54e71c
  

  dicentra13

  authored and

  Orbax Authors

  committed
  a54e71c

Commits on Dec 13, 2024

• Add type-registration mechanism for CheckpointHandlers.

  Show description for 6f41653
  

  BlaziusMaximus

  authored and

  Orbax Authors

  committed
  6f41653

Commits on Dec 12, 2024

• Internal

  Show description for 196d498
  

  liangyaning33

  authored and

  Orbax Authors

  committed
  196d498

• No public description

  Show description for 679a4a9
  

  liangyaning33

  authored and

  Orbax Authors

  committed
  679a4a9

Commits on Dec 11, 2024

• No public description

  Show description for b2b41e0
  

  liangyaning33

  authored and

  Orbax Authors

  committed
  b2b41e0

• No public description

  Show description for f057a24
  

  liangyaning33

  authored and

  Orbax Authors

  committed
  f057a24

• Add type-registration mechanism for CheckpointHandlers.

  Show description for 51e624a
  

  BlaziusMaximus

  authored and

  Orbax Authors

  committed
  51e624a

• Add type-registration mechanism for CheckpointHandlers.

  Show description for 6a34d19
  

  BlaziusMaximus

  authored and

  Orbax Authors

  committed
  6a34d19

Commits on Dec 10, 2024

• Fix logging error arising from None value.

  Show description for c65b985
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  c65b985

• Add Layout support in StandardCheckpointHandler

  Show description for 0c0216d
  

  ChromeHearts

  authored and

  Orbax Authors

  committed
  0c0216d

• Update CompositeCheckpointHandler.metadata() to return StepMetadata.

  Show description for 29cbf35
  

  BlaziusMaximus

  authored and

  Orbax Authors

  committed
  29cbf35

Commits on Dec 5, 2024

• Ignore not-exists and not-dir errors while building step metadata in _StandardNameFormat.

  Show description for ac2d276
  

  niketkumar

  authored and

  Orbax Authors

  committed
  ac2d276

Commits on Dec 4, 2024

• Internal changes

  Show description for 9b38ce3
  

  ChromeHearts

  authored and

  Orbax Authors

  committed
  9b38ce3

• fix a regression bug on deserialization with custom Layout

  Show description for 7ac1db1
  

  ChromeHearts

  authored and

  Orbax Authors

  committed
  7ac1db1

• Release orbax-checkpoint 0.10.2.

  Show description for d6101ba
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  d6101ba

• Add debug logging for restoration shardings in emergency checkpointing.

  Show description for 01d2e25
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  01d2e25

• Open source emergency checkpoint integration test script.

  Show description for 891da7b
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  891da7b

Commits on Dec 3, 2024

• Internal change.

  Show description for 68eec2e
  

  Orbax Authors

  committed
  68eec2e

• Removes the Tensorflow Module base class from the JaxModule.

  Show description for 4c02ff6
  

  Orbax Authors

  committed
  4c02ff6

• Avoid duplicate reads of the same index when loading.

  Show description for c7af62a
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  c7af62a

• Split out atomicity_types module from atomicity module so that TemporaryPath Protocol can be defined at a lower level than atomicity module.

  Show description for 9baf7db
  

  Orbax Authors

  committed
  9baf7db

Commits on Dec 2, 2024

• Removes the build function from the OrbaxModule and moves much of this functionality to the OrbaxExport to match the logic in TensorflowExport.

  Show description for ad1b199
  

  Orbax Authors

  committed
  ad1b199

Pagination

PreviousNext