orbax

orbax copied to clipboard

Adds a function to check the inputs have default values in orbax savedmodel.

copybara-service[bot]

[REX] Enable ttl based deletion of step and substeps

copybara-service[bot]

Adding selective e2e tests obm export tests

1

Adding selective e2e tests obm export tests

copybara-service[bot]

When a checkpoint folder is renamed, it's moved into a _todelete_ subdirectory. This directory is created as a sibling to the main checkpoints/ folder. This approach prevents these renamed folders...

deepikarajani24

Minor typo fix of the `resume` method docstring in standard_checkpointer.py

vfdev-5

internal

1

internal

copybara-service[bot]

Small refactoring: Pass exactly what the function needs

1

Small refactoring: Pass exactly what the function needs

copybara-service[bot]

Move implementation details of Orbax V1 `save_pytree` and `save_checkpointables` into helper file.

copybara-service[bot]

Add a new parameter `background_thread_count` to allow users to set the number of threads for ThreadedCheckpointDeleter.

copybara-service[bot]

Add an unit test for oex_orchestration.create().

1

Add an unit test for oex_orchestration.create().

copybara-service[bot]