Back to Search
python-orbax-checkpoint
Version 0.11.14-1 from cachyos (any)
Description
Common utility libraries for JAX users (checkpoint).
Homepage
https://github.com/google/orbax
License(s)
Apache-2.0
Build Date
Packager
CachyOS <admin@cachyos.org>
Package Size
982.06 KB
Installed Size
4.75 MB
Download Mirror
https://cdn77.cachyos.org/repo/x86_64/cachyos/python-orbax-checkpoint-0.11.14-1-any.pkg.tar.zst
SHA256 Sum
272faeb7b32b33120893af3ab76b26591c53d23b46c9f911b0b85c1a960a104b
Dependencies
python-abslpython-etilspython-humanizepython-jax>=0.5.0python-msgpackpython-nest-asynciopython-numpypython-protobufpython-simplejsonpython-tensorstorepython-typing_extensionspython-yaml
Optional Deps
python-orbax-export: Serialization JAX to TensorFlow Saved Models.
Provides
N/A
Conflicts With
python-orbax
Replaces
python-orbax
Package Files
usr/
usr/lib/
usr/lib/python3.13/
usr/lib/python3.13/site-packages/
usr/lib/python3.13/site-packages/orbax/
usr/lib/python3.13/site-packages/orbax/checkpoint/
usr/lib/python3.13/site-packages/orbax/checkpoint/BUILD
usr/lib/python3.13/site-packages/orbax/checkpoint/__init__.py
usr/lib/python3.13/site-packages/orbax/checkpoint/__pycache__/
usr/lib/python3.13/site-packages/orbax/checkpoint/__pycache__/__init__.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/__pycache__/abstract_checkpoint_manager.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/__pycache__/aggregate_handlers.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/__pycache__/args.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/__pycache__/arrays.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/__pycache__/checkpoint_args.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/__pycache__/checkpoint_args_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/__pycache__/checkpoint_manager.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/__pycache__/checkpoint_manager_options_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/__pycache__/checkpoint_managers.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/__pycache__/checkpoint_utils.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/__pycache__/checkpoint_utils_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/__pycache__/checkpointers.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/__pycache__/conftest.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/__pycache__/future.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/__pycache__/handlers.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/__pycache__/msgpack_utils.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/__pycache__/msgpack_utils_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/__pycache__/options.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/__pycache__/path.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/__pycache__/single_host_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/__pycache__/test_utils.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/__pycache__/test_utils_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/__pycache__/transform_utils.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/__pycache__/transform_utils_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/__pycache__/tree.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/__pycache__/type_handlers.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/__pycache__/utils.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/__pycache__/version.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/BUILD
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/README.md
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/__pycache__/
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/__pycache__/asyncio_utils.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/__pycache__/asyncio_utils_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/__pycache__/composite.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/__pycache__/composite_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/__pycache__/threading.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/arrays/
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/arrays/BUILD
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/arrays/__pycache__/
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/arrays/__pycache__/abstract_arrays.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/arrays/__pycache__/abstract_arrays_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/arrays/__pycache__/fragments.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/arrays/__pycache__/fragments_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/arrays/__pycache__/numpy_utils.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/arrays/__pycache__/numpy_utils_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/arrays/__pycache__/subchunking.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/arrays/__pycache__/subchunking_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/arrays/__pycache__/types.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/arrays/abstract_arrays.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/arrays/abstract_arrays_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/arrays/fragments.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/arrays/fragments_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/arrays/numpy_utils.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/arrays/numpy_utils_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/arrays/subchunking.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/arrays/subchunking_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/arrays/types.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/asyncio_utils.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/asyncio_utils_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/checkpoint_managers/
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/checkpoint_managers/BUILD
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/checkpoint_managers/__pycache__/
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/checkpoint_managers/__pycache__/policy_checkpoint_info.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/checkpoint_managers/__pycache__/preservation_policy.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/checkpoint_managers/__pycache__/preservation_policy_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/checkpoint_managers/__pycache__/save_decision_policy.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/checkpoint_managers/policy_checkpoint_info.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/checkpoint_managers/preservation_policy.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/checkpoint_managers/preservation_policy_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/checkpoint_managers/save_decision_policy.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/checkpointers/
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/checkpointers/BUILD
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/checkpointers/__pycache__/
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/checkpointers/__pycache__/abstract_checkpointer.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/checkpointers/__pycache__/async_checkpointer.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/checkpointers/__pycache__/checkpointer.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/checkpointers/__pycache__/checkpointer_test_utils.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/checkpointers/__pycache__/pytree_checkpointer.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/checkpointers/__pycache__/standard_checkpointer.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/checkpointers/abstract_checkpointer.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/checkpointers/async_checkpointer.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/checkpointers/checkpointer_test_utils.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/checkpointers/pytree_checkpointer.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/checkpointers/standard_checkpointer.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/composite.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/composite_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/futures/
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/futures/BUILD
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/futures/__pycache__/
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/futures/__pycache__/future.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/futures/__pycache__/signaling_client.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/futures/__pycache__/signaling_client_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/futures/__pycache__/synchronization.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/futures/__pycache__/synchronization_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/futures/future.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/futures/signaling_client.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/futures/signaling_client_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/futures/synchronization.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/futures/synchronization_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/BUILD
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/__pycache__/
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/__pycache__/array_checkpoint_handler.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/__pycache__/async_checkpoint_handler.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/__pycache__/base_pytree_checkpoint_handler.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/__pycache__/checkpoint_handler.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/__pycache__/composite_checkpoint_handler.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/__pycache__/composite_checkpoint_handler_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/__pycache__/handler_registration.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/__pycache__/handler_registration_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/__pycache__/handler_type_registry.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/__pycache__/handler_type_registry_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/__pycache__/json_checkpoint_handler.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/__pycache__/json_checkpoint_handler_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/__pycache__/proto_checkpoint_handler.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/__pycache__/pytree_checkpoint_handler.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/__pycache__/pytree_checkpoint_handler_test_utils.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/__pycache__/random_key_checkpoint_handler.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/__pycache__/random_key_checkpoint_handler_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/__pycache__/standard_checkpoint_handler.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/__pycache__/standard_checkpoint_handler_test_utils.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/array_checkpoint_handler.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/async_checkpoint_handler.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/checkpoint_handler.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/composite_checkpoint_handler.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/composite_checkpoint_handler_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/handler_registration.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/handler_registration_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/handler_type_registry.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/handler_type_registry_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/json_checkpoint_handler.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/json_checkpoint_handler_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/proto_checkpoint_handler.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/pytree_checkpoint_handler_test_utils.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/random_key_checkpoint_handler.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/random_key_checkpoint_handler_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/standard_checkpoint_handler.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/handlers/standard_checkpoint_handler_test_utils.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/logging/
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/logging/BUILD
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/logging/__pycache__/
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/logging/__pycache__/abstract_logger.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/logging/__pycache__/cloud_logger.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/logging/__pycache__/cloud_logger_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/logging/__pycache__/composite_logger.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/logging/__pycache__/composite_logger_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/logging/__pycache__/standard_logger.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/logging/__pycache__/standard_logger_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/logging/__pycache__/step_statistics.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/logging/abstract_logger.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/logging/cloud_logger.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/logging/cloud_logger_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/logging/composite_logger.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/logging/composite_logger_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/logging/standard_logger.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/logging/standard_logger_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/logging/step_statistics.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/BUILD
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/__pycache__/
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/__pycache__/array_metadata.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/__pycache__/array_metadata_store.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/__pycache__/array_metadata_store_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/__pycache__/checkpoint.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/__pycache__/checkpoint_info.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/__pycache__/checkpoint_info_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/__pycache__/checkpoint_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/__pycache__/empty_values.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/__pycache__/empty_values_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/__pycache__/metadata_serialization_utils.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/__pycache__/pytree_metadata_options.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/__pycache__/root_metadata_serialization.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/__pycache__/sharding.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/__pycache__/sharding_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/__pycache__/step_metadata_serialization.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/__pycache__/tree.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/__pycache__/tree_rich_types.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/__pycache__/tree_rich_types_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/__pycache__/tree_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/__pycache__/value.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/__pycache__/value_metadata_entry.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/array_metadata.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/array_metadata_store_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/checkpoint.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/checkpoint_info.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/checkpoint_info_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/checkpoint_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/empty_values.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/empty_values_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/metadata_serialization_utils.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/pytree_metadata_options.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/root_metadata_serialization.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/sharding.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/sharding_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/step_metadata_serialization.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/tree.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/tree_rich_types.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/tree_rich_types_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/tree_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/value.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/metadata/value_metadata_entry.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/multihost/
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/multihost/BUILD
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/multihost/__pycache__/
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/multihost/__pycache__/multihost.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/multihost/__pycache__/multislice.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/multihost/multihost.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/multihost/multislice.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/path/
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/path/BUILD
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/path/__init__.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/path/__pycache__/
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/path/__pycache__/__init__.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/path/__pycache__/async_utils.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/path/__pycache__/atomicity.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/path/__pycache__/atomicity_defaults.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/path/__pycache__/atomicity_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/path/__pycache__/atomicity_types.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/path/__pycache__/deleter.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/path/__pycache__/deleter_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/path/__pycache__/format_utils.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/path/__pycache__/format_utils_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/path/__pycache__/step.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/path/__pycache__/step_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/path/__pycache__/utils.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/path/async_utils.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/path/atomicity.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/path/atomicity_defaults.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/path/atomicity_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/path/atomicity_types.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/path/deleter.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/path/deleter_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/path/format_utils.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/path/format_utils_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/path/snapshot/
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/path/snapshot/BUILD
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/path/snapshot/__pycache__/
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/path/snapshot/__pycache__/snapshot.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/path/snapshot/__pycache__/snapshot_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/path/snapshot/snapshot.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/path/snapshot/snapshot_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/path/step.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/path/step_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/path/utils.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/serialization/
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/serialization/BUILD
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/serialization/__pycache__/
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/serialization/__pycache__/replica_slices.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/serialization/__pycache__/replica_slices_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/serialization/__pycache__/serialization.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/serialization/__pycache__/serialization_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/serialization/__pycache__/tensorstore_utils.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/serialization/__pycache__/tensorstore_utils_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/serialization/__pycache__/type_handlers.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/serialization/__pycache__/types.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/serialization/replica_slices.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/serialization/replica_slices_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/serialization/serialization.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/serialization/serialization_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/serialization/tensorstore_utils.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/serialization/tensorstore_utils_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/serialization/type_handlers.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/serialization/types.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/testing/
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/testing/BUILD
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/testing/__pycache__/
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/testing/__pycache__/multiprocess_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/testing/__pycache__/test_tree_utils.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/testing/multiprocess_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/testing/test_tree_utils.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/threading.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/tree/
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/tree/BUILD
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/tree/__pycache__/
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/tree/__pycache__/parts_of.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/tree/__pycache__/parts_of_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/tree/__pycache__/structure_utils.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/tree/__pycache__/structure_utils_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/tree/__pycache__/types.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/tree/__pycache__/utils.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/tree/__pycache__/utils_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/tree/parts_of.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/tree/parts_of_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/tree/structure_utils.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/tree/structure_utils_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/tree/types.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/tree/utils.py
usr/lib/python3.13/site-packages/orbax/checkpoint/_src/tree/utils_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/abstract_checkpoint_manager.py
usr/lib/python3.13/site-packages/orbax/checkpoint/aggregate_handlers.py
usr/lib/python3.13/site-packages/orbax/checkpoint/args.py
usr/lib/python3.13/site-packages/orbax/checkpoint/arrays.py
usr/lib/python3.13/site-packages/orbax/checkpoint/checkpoint_args.py
usr/lib/python3.13/site-packages/orbax/checkpoint/checkpoint_args_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/checkpoint_manager.py
usr/lib/python3.13/site-packages/orbax/checkpoint/checkpoint_manager_options_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/checkpoint_managers.py
usr/lib/python3.13/site-packages/orbax/checkpoint/checkpoint_utils.py
usr/lib/python3.13/site-packages/orbax/checkpoint/checkpoint_utils_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/checkpointers.py
usr/lib/python3.13/site-packages/orbax/checkpoint/conftest.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/README.md
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/__init__.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/__pycache__/
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/__pycache__/__init__.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/BUILD
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/__init__.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/__pycache__/
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/__pycache__/__init__.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/__pycache__/broadcast_multislice_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/__pycache__/checkpoint_manager.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/__pycache__/checkpoint_manager_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/__pycache__/integration_test_utils.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/__pycache__/local_checkpoint_data_debugging.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/__pycache__/local_checkpoint_data_debugging_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/__pycache__/local_checkpoint_manager_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/__pycache__/mesh_consistency.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/__pycache__/multihost.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/__pycache__/multihost_simulated_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/__pycache__/multihost_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/__pycache__/process_metadata_checkpoint_handler.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/__pycache__/replicator_checkpoint_manager.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/__pycache__/replicator_checkpoint_manager_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/__pycache__/single_slice_checkpoint_manager_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/__pycache__/test_utils.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/broadcast_multislice_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/checkpoint_manager.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/checkpoint_manager_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/integration_test_utils.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/local_checkpoint_data_debugging.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/local_checkpoint_data_debugging_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/local_checkpoint_manager_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/mesh_consistency.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/multihost.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/multihost_simulated_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/multihost_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/process_metadata_checkpoint_handler.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/replicator_checkpoint_manager.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/replicator_checkpoint_manager_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/single_slice_checkpoint_manager_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/test_scripts/
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/test_scripts/__pycache__/
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/test_scripts/__pycache__/test_emergency_checkpointing.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/test_scripts/test_emergency_checkpointing.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/emergency/test_utils.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/__init__.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/__pycache__/
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/__pycache__/__init__.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/__pycache__/handlers.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/__pycache__/training.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/context/
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/context/BUILD
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/context/__pycache__/
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/context/__pycache__/context.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/context/__pycache__/context_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/context/__pycache__/options.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/context/context.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/context/context_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/context/options.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/handlers/
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/handlers/BUILD
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/handlers/__pycache__/
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/handlers/__pycache__/compatibility.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/handlers/__pycache__/composite_handler.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/handlers/__pycache__/composite_handler_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/handlers/__pycache__/global_registration.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/handlers/__pycache__/json_handler.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/handlers/__pycache__/json_handler_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/handlers/__pycache__/proto_handler.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/handlers/__pycache__/pytree_handler.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/handlers/__pycache__/pytree_handler_test_base.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/handlers/__pycache__/registration.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/handlers/__pycache__/registration_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/handlers/__pycache__/stateful_checkpointable_handler.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/handlers/__pycache__/types.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/handlers/__pycache__/types_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/handlers/compatibility.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/handlers/composite_handler.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/handlers/composite_handler_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/handlers/global_registration.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/handlers/json_handler.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/handlers/json_handler_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/handlers/proto_handler.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/handlers/pytree_handler.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/handlers/pytree_handler_test_base.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/handlers/registration.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/handlers/registration_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/handlers/stateful_checkpointable_handler.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/handlers/types.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/handlers/types_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/loading/
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/loading/BUILD
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/loading/__pycache__/
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/loading/__pycache__/loading.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/loading/loading.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/metadata/
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/metadata/BUILD
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/metadata/__pycache__/
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/metadata/__pycache__/loading.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/metadata/__pycache__/loading_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/metadata/__pycache__/serialization.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/metadata/__pycache__/types.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/metadata/loading.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/metadata/loading_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/metadata/serialization.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/metadata/types.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/path/
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/path/BUILD
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/path/__pycache__/
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/path/__pycache__/async_utils.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/path/__pycache__/async_utils_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/path/__pycache__/format_utils.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/path/__pycache__/format_utils_test.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/path/__pycache__/step.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/path/__pycache__/types.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/path/async_utils.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/path/async_utils_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/path/format_utils.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/path/format_utils_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/path/step.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/path/types.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/saving/
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/saving/BUILD
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/saving/__pycache__/
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/saving/__pycache__/saving.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/saving/saving.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/serialization/
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/serialization/BUILD
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/serialization/__pycache__/
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/serialization/__pycache__/array_leaf_handler.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/serialization/__pycache__/compatibility.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/serialization/__pycache__/numpy_leaf_handler.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/serialization/__pycache__/registration.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/serialization/__pycache__/types.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/serialization/array_leaf_handler.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/serialization/compatibility.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/serialization/numpy_leaf_handler.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/serialization/registration.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/serialization/types.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/synchronization/
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/synchronization/BUILD
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/synchronization/__pycache__/
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/synchronization/__pycache__/multihost.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/synchronization/__pycache__/signaling_client.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/synchronization/__pycache__/synchronization.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/synchronization/__pycache__/types.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/synchronization/multihost.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/synchronization/signaling_client.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/synchronization/synchronization.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/synchronization/types.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/testing/
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/testing/BUILD
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/testing/__pycache__/
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/testing/__pycache__/array_utils.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/testing/__pycache__/handler_utils.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/testing/__pycache__/path_utils.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/testing/__pycache__/save_load_test_base.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/testing/__pycache__/tree_utils.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/testing/__pycache__/v0v1_compatibility_save_load_test_base.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/testing/array_utils.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/testing/google/
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/testing/google/BUILD
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/testing/handler_utils.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/testing/path_utils.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/testing/pathways/
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/testing/pathways/BUILD
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/testing/save_load_test_base.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/testing/tree_utils.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/testing/v0v1_compatibility_save_load_test_base.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/training/
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/training/BUILD
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/training/__pycache__/
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/training/__pycache__/checkpointer.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/training/__pycache__/checkpointer_test_base.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/training/__pycache__/errors.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/training/__pycache__/preservation_policies.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/training/__pycache__/save_decision_policies.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/training/__pycache__/v0v1_compatibility_checkpointer_test_base.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/training/checkpointer.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/training/checkpointer_test_base.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/training/errors.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/training/metadata/
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/training/metadata/BUILD
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/training/metadata/__pycache__/
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/training/metadata/__pycache__/types.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/training/metadata/types.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/training/pathways/
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/training/pathways/BUILD
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/training/preservation_policies.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/training/save_decision_policies.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/training/v0v1_compatibility_checkpointer_test_base.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/tree/
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/tree/BUILD
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/tree/__pycache__/
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/tree/__pycache__/types.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/_src/tree/types.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/handlers.py
usr/lib/python3.13/site-packages/orbax/checkpoint/experimental/v1/training.py
usr/lib/python3.13/site-packages/orbax/checkpoint/future.py
usr/lib/python3.13/site-packages/orbax/checkpoint/handlers.py
usr/lib/python3.13/site-packages/orbax/checkpoint/logging/
usr/lib/python3.13/site-packages/orbax/checkpoint/logging/__init__.py
usr/lib/python3.13/site-packages/orbax/checkpoint/logging/__pycache__/
usr/lib/python3.13/site-packages/orbax/checkpoint/logging/__pycache__/__init__.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/logging/__pycache__/step_statistics.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/logging/step_statistics.py
usr/lib/python3.13/site-packages/orbax/checkpoint/metadata/
usr/lib/python3.13/site-packages/orbax/checkpoint/metadata/BUILD
usr/lib/python3.13/site-packages/orbax/checkpoint/metadata/__init__.py
usr/lib/python3.13/site-packages/orbax/checkpoint/metadata/__pycache__/
usr/lib/python3.13/site-packages/orbax/checkpoint/metadata/__pycache__/__init__.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/metadata/__pycache__/tree.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/metadata/__pycache__/value.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/metadata/tree.py
usr/lib/python3.13/site-packages/orbax/checkpoint/metadata/value.py
usr/lib/python3.13/site-packages/orbax/checkpoint/msgpack_utils.py
usr/lib/python3.13/site-packages/orbax/checkpoint/msgpack_utils_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/multihost/
usr/lib/python3.13/site-packages/orbax/checkpoint/multihost/BUILD
usr/lib/python3.13/site-packages/orbax/checkpoint/multihost/__init__.py
usr/lib/python3.13/site-packages/orbax/checkpoint/multihost/__pycache__/
usr/lib/python3.13/site-packages/orbax/checkpoint/multihost/__pycache__/__init__.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/multihost/__pycache__/multislice.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/multihost/multislice.py
usr/lib/python3.13/site-packages/orbax/checkpoint/options.py
usr/lib/python3.13/site-packages/orbax/checkpoint/path.py
usr/lib/python3.13/site-packages/orbax/checkpoint/proto/
usr/lib/python3.13/site-packages/orbax/checkpoint/proto/testing/
usr/lib/python3.13/site-packages/orbax/checkpoint/proto/testing/BUILD
usr/lib/python3.13/site-packages/orbax/checkpoint/serialization/
usr/lib/python3.13/site-packages/orbax/checkpoint/serialization/BUILD
usr/lib/python3.13/site-packages/orbax/checkpoint/serialization/__init__.py
usr/lib/python3.13/site-packages/orbax/checkpoint/serialization/__pycache__/
usr/lib/python3.13/site-packages/orbax/checkpoint/serialization/__pycache__/__init__.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/serialization/__pycache__/tensorstore_utils.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/serialization/tensorstore_utils.py
usr/lib/python3.13/site-packages/orbax/checkpoint/single_host_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/test_utils.py
usr/lib/python3.13/site-packages/orbax/checkpoint/test_utils_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/testing/
usr/lib/python3.13/site-packages/orbax/checkpoint/testing/BUILD
usr/lib/python3.13/site-packages/orbax/checkpoint/testing/__init__.py
usr/lib/python3.13/site-packages/orbax/checkpoint/testing/__pycache__/
usr/lib/python3.13/site-packages/orbax/checkpoint/testing/__pycache__/__init__.cpython-313.opt-1.pyc
usr/lib/python3.13/site-packages/orbax/checkpoint/testing/aggregate_all.checkpoint/
usr/lib/python3.13/site-packages/orbax/checkpoint/testing/aggregate_all.checkpoint/_METADATA
usr/lib/python3.13/site-packages/orbax/checkpoint/testing/aggregate_all.checkpoint/checkpoint
usr/lib/python3.13/site-packages/orbax/checkpoint/testing/aggregate_some.checkpoint/
usr/lib/python3.13/site-packages/orbax/checkpoint/testing/aggregate_some.checkpoint/_METADATA
usr/lib/python3.13/site-packages/orbax/checkpoint/testing/aggregate_some.checkpoint/a/
usr/lib/python3.13/site-packages/orbax/checkpoint/testing/aggregate_some.checkpoint/a/.zarray
usr/lib/python3.13/site-packages/orbax/checkpoint/testing/aggregate_some.checkpoint/a/0
usr/lib/python3.13/site-packages/orbax/checkpoint/testing/aggregate_some.checkpoint/b/
usr/lib/python3.13/site-packages/orbax/checkpoint/testing/aggregate_some.checkpoint/b/.zarray
usr/lib/python3.13/site-packages/orbax/checkpoint/testing/aggregate_some.checkpoint/b/0
usr/lib/python3.13/site-packages/orbax/checkpoint/testing/aggregate_some.checkpoint/checkpoint
usr/lib/python3.13/site-packages/orbax/checkpoint/transform_utils.py
usr/lib/python3.13/site-packages/orbax/checkpoint/transform_utils_test.py
usr/lib/python3.13/site-packages/orbax/checkpoint/tree.py
usr/lib/python3.13/site-packages/orbax/checkpoint/type_handlers.py
usr/lib/python3.13/site-packages/orbax/checkpoint/utils.py
usr/lib/python3.13/site-packages/orbax/checkpoint/version.py
usr/lib/python3.13/site-packages/orbax_checkpoint-0.11.14.dist-info/
usr/lib/python3.13/site-packages/orbax_checkpoint-0.11.14.dist-info/METADATA
usr/lib/python3.13/site-packages/orbax_checkpoint-0.11.14.dist-info/RECORD
usr/lib/python3.13/site-packages/orbax_checkpoint-0.11.14.dist-info/WHEEL
usr/lib/python3.13/site-packages/orbax_checkpoint-0.11.14.dist-info/licenses/
usr/lib/python3.13/site-packages/orbax_checkpoint-0.11.14.dist-info/licenses/LICENSE
usr/share/
usr/share/doc/
usr/share/doc/python-orbax-checkpoint/
usr/share/doc/python-orbax-checkpoint/README.md