Deepika Rajani

Summary: This PR decouples Orbax from direct GCS API dependencies by delegating all filesystem operations to the underlying TensorFlow gfile implementation. This change improves testability and makes Orbax less susceptible...

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...

Updated hns enabled check

## Summary This PR updates the TensorFlow and Orbax dependencies to support the new HNS-native `RenameFolder` API. ## Changes To leverage this feature for HNS buckets, you'd need to configure...

Support empty folder deletion for HNS bucket with Orbax

1

Orbax made a change to support empty folders deletion which require enable_hns_rmtree flag to be True in CheckpointManagerOptions. - Tested orbax regular checkpoint with HNS bucket with keep_last_n as 3,...

gcsfuse_mount_spec parsing and supporting mount_options from command line

1

This PR improves the command-line parsing for --gcsfuse_mount_spec to allow for multiple key-value pairs and robustly handle custom GCS FUSE mount_options. **Usage** 1. To specify custom mount_options: The entire argument...