orbax-checkpoint

View on PyPIReverse Dependencies (24)

0.11.1 orbax_checkpoint-0.11.1-py3-none-any.whl

Wheel Details

Project: orbax-checkpoint
Version: 0.11.1
Filename: orbax_checkpoint-0.11.1-py3-none-any.whl
Download: [link]
Size: 394885
MD5: 607a6fb773f46e88d652bf0ff700a643
SHA256: 9cfbe49331ffdef4aedfec337da683eb8fc65221dc43924cf6daea6c27d6dc06
Uploaded: 2025-01-28 23:09:18 +0000

dist-info

METADATA

Metadata-Version: 2.3
Name: orbax-checkpoint
Version: 0.11.1
Summary: Orbax Checkpoint
Author-Email: Orbax Authors <orbax-dev[at]google.com>
Project-Url: homepage, http://github.com/google/orbax
Project-Url: repository, http://github.com/google/orbax
Keywords: JAX machine learning,checkpoint,training
Classifier: Development Status :: 4 - Beta
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Programming Language :: Python
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Python: >=3.10
Requires-Dist: absl-py
Requires-Dist: etils[epath,epy]
Requires-Dist: typing_extensions
Requires-Dist: msgpack
Requires-Dist: jax (>=0.4.34)
Requires-Dist: numpy
Requires-Dist: pyyaml
Requires-Dist: tensorstore (>=0.1.68)
Requires-Dist: nest_asyncio
Requires-Dist: protobuf
Requires-Dist: humanize
Requires-Dist: simplejson (>=3.16.0)
Requires-Dist: google-cloud-logging; extra == "testing"
Requires-Dist: mock; extra == "testing"
Requires-Dist: flax; extra == "testing"
Requires-Dist: pytest; extra == "testing"
Requires-Dist: pytest-xdist; extra == "testing"
Requires-Dist: chex; extra == "testing"
Provides-Extra: testing
Description-Content-Type: text/markdown
[Description omitted; length: 557 characters]

WHEEL

Wheel-Version: 1.0
Generator: flit 3.10.1
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
orbax/checkpoint/BUILD sha256=bGIeeFEHqF1kPHn8AkPa2YXF7ubulL56cAu698YqBtc 12111
orbax/checkpoint/__init__.py sha256=L8I5AN_6ieCmWW7diphEDuyJm52I40a5_3LlOJ4jX-I 4004
orbax/checkpoint/abstract_checkpoint_manager.py sha256=F9onWioE7KpBEHfReEyVvz5BMVfESrn4ldSI2jCPhiI 11853
orbax/checkpoint/aggregate_handlers.py sha256=d8BFUyMI0T4DEG8oASZHtnt1rwJ8FMnEaeHD8p5ZXT4 3526
orbax/checkpoint/args.py sha256=f7N-9C7A7iiC-CWC31-VqFu4cLh78nib0t4o9kgB3Mo 3112
orbax/checkpoint/arrays.py sha256=10B3noHVfQFhho_vLLQW0mDElfcFKqQ33gTcTBS1qyg 870
orbax/checkpoint/checkpoint_args.py sha256=6zfNj1A00DpaZpIPzERanPnFjMS0RZJ4OtcgIocENWU 5533
orbax/checkpoint/checkpoint_args_test.py sha256=VU1WgkswD1UwZwWZOvXmdfVIU0u37XKpp-y-8JBcbdk 5743
orbax/checkpoint/checkpoint_manager.py sha256=Msbup3-YvNTGw9OzeEXsMtvToUWS-Jpkcml0KfIzKjU 74164
orbax/checkpoint/checkpoint_managers.py sha256=2ruw7rIblVvT5VLO-QwXJzB7ZY-EUrLvAJyboxf0BM8 1272
orbax/checkpoint/checkpoint_utils.py sha256=hMrt1YbjKymcn9CojvN0TXMFqLtJyypI0xYX3kh7PXk 19115
orbax/checkpoint/checkpoint_utils_test.py sha256=_5F1VlUdru6Oks95hNMCjgySN-Wdq5gYsuCT8vAE3oI 14911
orbax/checkpoint/checkpointers.py sha256=o6VsCHiZLZiY_TQV0fM1BDs_uMVdJeFpZXGLVbq3usg 1151
orbax/checkpoint/conftest.py sha256=ghkxmpxGtLnqqD7-UeNmnpTo9didMaWa1Q9sDHnhf4Y 740
orbax/checkpoint/future.py sha256=fpEmakfuL1oP5P5kfndp2NMhBjVkYoUAmcEoxVqUJTo 935
orbax/checkpoint/handlers.py sha256=ByZn5pzB8OQTDNAX7TM4s7ITt_MHQ0uYkKcAL_rANWw 3761
orbax/checkpoint/logging.py sha256=PX8PtWrde6ExAE7Erv36hefpy7jClISakgoqKxddyjo 1071
orbax/checkpoint/msgpack_utils.py sha256=72Eb3h3Fy6kTkU39p2sAKSxCiBXbSLJBZP7ts6nACbY 7673
orbax/checkpoint/msgpack_utils_test.py sha256=h8I8Fp42gK02GqeexVuH-TB54c4M5_xrWxsa_ytOkiQ 1129
orbax/checkpoint/options.py sha256=Obv-N3ArJOs08CZb-ZAi0JPBwih4ZCDS_UvGO4AlwMw 2699
orbax/checkpoint/path.py sha256=2v6-ysjcsdDQw6s2fPWgtMfsL4jdceGPYuiNy6PZlBg 1040
orbax/checkpoint/single_host_test.py sha256=tVJW4oq-nTlb0pgYzsYxmIpHaW-KeV_WJUxyXecIT6I 8056
orbax/checkpoint/test_utils.py sha256=89QvnIkpbZOaGdyoX9kLenEOA7gKRtL7NwSwDdC3WDk 22506
orbax/checkpoint/test_utils_test.py sha256=wJ8tltrx3AcrEXxk_KyHHzQsMaJ8KIbccTOxKg325yA 1374
orbax/checkpoint/transform_utils.py sha256=oPE_DMgZjpGT0k8EovCzhklFxED2u19VJFvjkcqM3HU 12807
orbax/checkpoint/transform_utils_test.py sha256=lLavzkCtjLM5KGlZWyucrSSvM7Kd8M7E-tDEPjScjGA 15213
orbax/checkpoint/tree.py sha256=V0LnFZ0B3TgmRBYkwj7TgNOdAoJzOQpIfo8gi262SNs 997
orbax/checkpoint/type_handlers.py sha256=B41Mqn2dRwWLqf0-Wc0p1N7wimJEI40oXDCzSXdG4oU 3220
orbax/checkpoint/utils.py sha256=cgUx0uRrNEpoi4uoyd1vl6BCd1NeTcdPQP8aH-NnyZ8 6794
orbax/checkpoint/version.py sha256=R5fMAppJn0S0imlgpYNM1nfCV2M21K9I8QIVXQJq3a4 984
orbax/checkpoint/_src/BUILD sha256=MH3QeFuPOgqiuI8zS2rCu5bqhGqwFRBw-Z1tBH1RGIA 414
orbax/checkpoint/_src/README.md sha256=y1gWZ2_EHkGviJfTn9tEhhDM-BkKAXbhd1L9Qh7YXWU 300
orbax/checkpoint/_src/asyncio_utils.py sha256=zBWVIugHigzMs5X8KE732VKV7fOg737H9k7cltFJ8Nw 1508
orbax/checkpoint/_src/asyncio_utils_test.py sha256=ZZK4jvVW-8gqD4W4sR_a9D4mlL2PDJ9QAnEjs-Ury-I 8378
orbax/checkpoint/_src/composite.py sha256=1z4PIyIY1uE9q9F6kH9TrPiUS1Gz9ZSMbd90fnTbuwM 3286
orbax/checkpoint/_src/composite_test.py sha256=wGAxqFaptKmkBG0gM--J_iy3_xdObZzIkNsQCsdFuzU 3456
orbax/checkpoint/_src/arrays/BUILD sha256=p5TI0rtn_iUfSccs0_pWDfzzRnsn0X6Vm7dAAgxm1C4 1515
orbax/checkpoint/_src/arrays/abstract_arrays.py sha256=nkymKioKz4OwVlRXQha1LIPZmf7cBJy7r81WG-MQKS4 3097
orbax/checkpoint/_src/arrays/abstract_arrays_test.py sha256=rua0qvYez0Qo1oi5Lgvh7bmg9HpTfRwGzGGZ9_c4ii8 3972
orbax/checkpoint/_src/arrays/fragments.py sha256=7EPiN4pPBCzVO6WwCc3mmZkG331WK8g2AOUvbmAsMgw 9697
orbax/checkpoint/_src/arrays/fragments_test.py sha256=Uv19p7J6YCe1-BrYj8ceiW79WMLEur42PBCMoppSoAc 13580
orbax/checkpoint/_src/arrays/numpy_utils.py sha256=DrVLXTxTcWVvJK2JaL0Lq5O4PQCzAAKy_ZYT_rJr0t0 5221
orbax/checkpoint/_src/arrays/numpy_utils_test.py sha256=B6uPK1EC0IJ3g_FN7i0zvj2I5hqZd3Y4r-7kHzu1SLQ 3480
orbax/checkpoint/_src/arrays/subchunking.py sha256=Lsug6cs2F4luC_YmGlmwJUyRUeEQJCSjKgwxbaECJZ0 10314
orbax/checkpoint/_src/arrays/subchunking_test.py sha256=kNKDHovVi4Ta71IybG9YNzYsfyv2fszuQ2Fme2I5D6U 20346
orbax/checkpoint/_src/arrays/types.py sha256=LdKPpzxxkLJUIWtF1uaLGVFsJJQS_rTNICNWP43PVyg 1518
orbax/checkpoint/_src/checkpoint_managers/BUILD sha256=iwd4u_cher2kYROYumy1bLb8q-oJSYY1xyJcUbfQoX0 144
orbax/checkpoint/_src/checkpoint_managers/save_decision_policy.py sha256=7LofeHfrGPS3tze9NLnuM1nKw6tdno9XIXVnqB5OYtc 3771
orbax/checkpoint/_src/checkpointers/BUILD sha256=P4Z_8TqO5hYoAErX3WLr-8xIz1Cys4-S0lCtFZJVov0 2711
orbax/checkpoint/_src/checkpointers/abstract_checkpointer.py sha256=kXtf7ARoaJPppaQFqVjvd7sB7WMR7ovOY3Q-c96PyZM 3015
orbax/checkpoint/_src/checkpointers/async_checkpointer.py sha256=YVTYt3Me_j7AWF49Fe55oyk9ieDWHpIVNhuml4ZtGt0 17732
orbax/checkpoint/_src/checkpointers/checkpointer.py sha256=qNjE86EPY_AQO1P2oCT26AyjOhay3Lm_sAsW5J0gr5w 12299
orbax/checkpoint/_src/checkpointers/pytree_checkpointer.py sha256=f4pXIkNTVeE7mdDEKF5kgRDGORLwnGpStIzidLPNM2g 1451
orbax/checkpoint/_src/checkpointers/standard_checkpointer.py sha256=cEds3DDjJmF4Bpye_3t4UInyAhhVIiCtfMr539GBNOQ 5455
orbax/checkpoint/_src/futures/BUILD sha256=GLmDb7pWKsogCBaF8sI3qfUvddTFWMztnRG7MCi-tC4 480
orbax/checkpoint/_src/futures/future.py sha256=lYeM7amx8yn1o0jyWJvXAJlCOKM_n_c1_0WBhsrb7jE 12485
orbax/checkpoint/_src/futures/synchronization.py sha256=HCnywTZu00o1c3g3vwcelIk8qhvWjNgl6QeenKrjFRo 2317
orbax/checkpoint/_src/futures/synchronization_test.py sha256=MYR4gteyS7btDhYWgu_01k77MM66eMdCvqr_yowXFB4 1655
orbax/checkpoint/_src/handlers/BUILD sha256=w3dOd4tEOC-88YaecYqmiDfkzdRChYiL0Ip9LsJRoew 9422
orbax/checkpoint/_src/handlers/array_checkpoint_handler.py sha256=eZgGFaOiTtpe_fVpK67UpwchwSGO9qP2gG1veZ-6EAo 7199
orbax/checkpoint/_src/handlers/async_checkpoint_handler.py sha256=f5MsciYTbdksAZ_Cda6jHIhdBiZbKMa1SFT9FoaVGgU 1405
orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py sha256=yD8iQk60taMOTQDth7cP58eO1w7R2ufYptRq0cjNqZs 37976
orbax/checkpoint/_src/handlers/checkpoint_handler.py sha256=ihHJ9TMsMIjfdUjKNFsilmVrhvFtcq8-awkxB2xg1Uc 2479
orbax/checkpoint/_src/handlers/composite_checkpoint_handler.py sha256=4HkoJnf7e1C22yVg1fsp8A9w8AGW1I9e2gK4jcTrgWg 38863
orbax/checkpoint/_src/handlers/composite_checkpoint_handler_test.py sha256=mDR8lMHjFUREcmJ4m0PHLp7kXi2N_aZYhyKD8pxjrUk 36039
orbax/checkpoint/_src/handlers/handler_registration.py sha256=X-QlEIgx_8CTaVGcAAZApHME3GCPgyt9TvI2x4fPX9M 9366
orbax/checkpoint/_src/handlers/handler_registration_test.py sha256=_t8f9dTr9pWlMyuOYiIRTZjAeqfoDBMMfy2i1lI4Y2g 6984
orbax/checkpoint/_src/handlers/handler_type_registry.py sha256=PhsTLeVTrNh8XZnz1M2kwwtUmrzC55ZKW_Qw_yCosII 4059
orbax/checkpoint/_src/handlers/handler_type_registry_test.py sha256=xF9vbEw_BrmoRXy9NTj_ZrNr7WyMxMaljfjW6GAzgLk 7273
orbax/checkpoint/_src/handlers/json_checkpoint_handler.py sha256=eJqY2EY0ZyMRG2qdoIWsJlQRjci5p6WLtTzWvtRgf6k 4353
orbax/checkpoint/_src/handlers/json_checkpoint_handler_test.py sha256=A-JywejoxNRA3NdPzOvyYHSIdieB_9jChjZaSECjSRg 3168
orbax/checkpoint/_src/handlers/proto_checkpoint_handler.py sha256=O8QACeThxwwMQ9KuhG0IomaXw7B-SjrYWHeNo7J_8hA 4116
orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py sha256=az6Zo0VNFDy38ZUYulX6PhWtBhHdc8c-F7Ud25WJi28 43717
orbax/checkpoint/_src/handlers/random_key_checkpoint_handler.py sha256=4DnwDyc0F24VbPtipEHCnRWiM4-8y0fKkQpJvxF4QCY 9066
orbax/checkpoint/_src/handlers/random_key_checkpoint_handler_test.py sha256=jiL3-h-qjhxr2pGHRj9PwrVdzXETlKRF4MMDYK_DfrI 6633
orbax/checkpoint/_src/handlers/standard_checkpoint_handler.py sha256=feKqOHQNNC28_FuuWKiaUvcKrV2P7jYTDvwiUbrOkso 11429
orbax/checkpoint/_src/handlers/standard_checkpoint_handler_test_utils.py sha256=UGajpY3GxT0EKu-rMOuLq7ULy2S8V7uAHkNq7YYzk8I 14457
orbax/checkpoint/_src/logging/BUILD sha256=njcYdJvrQUUfRDFgln0qrDUFYgKddWFpuAgWeEVXnOA 999
orbax/checkpoint/_src/logging/abstract_logger.py sha256=8u8xe5Cv6Xg-WmRu6KVsX7auw34s2R0xD0XUnPXf5xE 1068
orbax/checkpoint/_src/logging/cloud_logger.py sha256=izydLNgRZ3MGskUusrawOrOwgShCyUyiXscdOW0-Gus 2584
orbax/checkpoint/_src/logging/cloud_logger_test.py sha256=nVfPOKGdDsDuCizXo7yF5wS9z_JH7LWV51u6zQXiG3o 1379
orbax/checkpoint/_src/logging/composite_logger.py sha256=FvdGg7IsqfC8JRg388XBm30kU1aQhARf-xw2n3pnrN0 1834
orbax/checkpoint/_src/logging/composite_logger_test.py sha256=cU5VJXbeP0qPfu98MZY608lpl9hvId2jKdfqAKVWBBs 3085
orbax/checkpoint/_src/logging/standard_logger.py sha256=wwJs2jwWGL6N3BY6y2DyUD3Dnc5aoHzxqhHtpXwGxdA 1214
orbax/checkpoint/_src/logging/standard_logger_test.py sha256=X751pltK0Yb5oaNRBg-jc_buJeVHcbpoz-BHwXKsGsQ 1185
orbax/checkpoint/_src/logging/step_statistics.py sha256=c5PxF3I7mGXxoRkeqMJeyQ_8Fk2I-BYmtEW__9e7vbg 5467
orbax/checkpoint/_src/metadata/BUILD sha256=PDcUuL8ftZSWIk8k09-zVgk7nvH9X_0qvz-Y9xRSswc 4384
orbax/checkpoint/_src/metadata/array_metadata.py sha256=iLNCVzdGYhr8ruEWlC21nRGtdWYk-FnixAelNpemJa4 1375
orbax/checkpoint/_src/metadata/array_metadata_store.py sha256=EIRAG1oFf3MjNqDJ1ra_gx7-uLB6N_tDDNBXJE9hf_Y 17936
orbax/checkpoint/_src/metadata/array_metadata_store_test.py sha256=CjrGWijG8wu1rCvh8P9YPTRI5jN7RMiFiiNpH3YrUE8 12400
orbax/checkpoint/_src/metadata/checkpoint.py sha256=F7E-hhSPZ_ETkUphk1_Vr39sxKHQNLKrL-tqFIg4iGU 15724
orbax/checkpoint/_src/metadata/checkpoint_test.py sha256=Tn3U7hKLsd-YNx523cnLfQzrk_5F8PDWe88L7q8V37E 23618
orbax/checkpoint/_src/metadata/empty_values.py sha256=Wjk4U62T_-IAucPl9_tT0uubRvnLBXOGH4AxuIh2w08 4062
orbax/checkpoint/_src/metadata/empty_values_test.py sha256=yxu5pDZu9ml8ESLpxXeD2Ect8E4DujoooljQEQjARig 9593
orbax/checkpoint/_src/metadata/metadata_serialization_utils.py sha256=JYpFUj9t6h3iNCcf06dbUgPQ20UL4kFtwCJ_Z36HaYg 1105
orbax/checkpoint/_src/metadata/pytree_metadata_options.py sha256=UgNRGHZPVw5YO_NV5LQKml-_fWNZIKc4A1iG2f7Mcqo 1235
orbax/checkpoint/_src/metadata/root_metadata_serialization.py sha256=HQJK9P8-h8VpqEe4shL23JmLNE8PA2NTN07D1n6yPn8 2110
orbax/checkpoint/_src/metadata/sharding.py sha256=2wDrBM5J4h83_PnTawVDuOG2bh-_4HL0IRvS4SycbZo 13086
orbax/checkpoint/_src/metadata/sharding_test.py sha256=3w4FYujGSGD45yhEBKtZK_reaDb0BTS7ThXP_sIQZWE 8964
orbax/checkpoint/_src/metadata/step_metadata_serialization.py sha256=W3l6kJHUKTfapCsQGl0HtIchyLf0u9-vavmzOV9vuZs 8211
orbax/checkpoint/_src/metadata/tree.py sha256=ylh8zBhhlPiPiNhz-Z9qTOooAzbHPcmcN7pWZHSzNB4 28989
orbax/checkpoint/_src/metadata/tree_rich_types.py sha256=e5bTPOcMnk2GJ7IFUMLSdqNf-AvNnzW-9BuOO-NqdOQ 7694
orbax/checkpoint/_src/metadata/tree_rich_types_test.py sha256=FQblQ8V6yXBg9bKxVMzMetmslTwzI8E8EDRonYwaoFM 1239
orbax/checkpoint/_src/metadata/tree_test.py sha256=qsB3Wyv99RxS8IGUhm1H__MO4Dy7gOWZLCiJMlj2NKA 12467
orbax/checkpoint/_src/metadata/value.py sha256=OIsGZKK0AEENjHdQOmL86jthb2R3ukrTQPwvSrtup_o 3755
orbax/checkpoint/_src/metadata/value_metadata_entry.py sha256=LJtTlkhXTiTJR_EuBjbSmo90iV9oohBMFHl-ewTkoCg 2989
orbax/checkpoint/_src/multihost/BUILD sha256=3JuVAaWdl_-Y1-SZLzxFNrlzpoz8YtE4yLcxOUJG0WM 219
orbax/checkpoint/_src/multihost/multihost.py sha256=YmEp_QgDfTSZW5h0Bt-szO8tBL9g-vkqcbanU6Giiy8 12896
orbax/checkpoint/_src/multihost/multislice.py sha256=7TbE1CoB7E97QjiS8eRhH1ImEq2kSsDc9V6sUsS3c_w 10318
orbax/checkpoint/_src/path/BUILD sha256=1TV58f0eStwcRKN3JO7_EpqhEg9lUoBZcT0bT-c_zqE 3349
orbax/checkpoint/_src/path/__init__.py sha256=9xDrETQE_Fq-P-NVacRNKcrVOEdoltsbm7FIRTl5lqc 649
orbax/checkpoint/_src/path/async_utils.py sha256=gRYq_mxwQVSU2_Ij5QPNajYo9rTTOP49StDBHwRpVRQ 1736
orbax/checkpoint/_src/path/atomicity.py sha256=EGzy2IDt_Smhn1zTL25sg-1zxt6-v5hgwoaFV5PnnEM 18549
orbax/checkpoint/_src/path/atomicity_defaults.py sha256=sh0Y-whv2PZSCdLb8Hh6eUyANH-2iLColTVXH_0PkJc 1733
orbax/checkpoint/_src/path/atomicity_test.py sha256=gODgf19lqaQ2r4M17zmTb1hgAhEnHmnN3RCCy16b-SI 5113
orbax/checkpoint/_src/path/atomicity_types.py sha256=tZs_Eu8WbITpQo_ZQzXgemPDmmB3qaDrBhvek5IstWg 2927
orbax/checkpoint/_src/path/deleter.py sha256=gMD7IaWODumcyLn2qOm3WwNODS_LtBmfGqqLyes9QPQ 6279
orbax/checkpoint/_src/path/deleter_test.py sha256=gpeFreqcQMbsjH0OVMlEdPWWMpewaj-IWXkAUcmkyxk 2095
orbax/checkpoint/_src/path/format_utils.py sha256=rBaLhMeZoa3xGE-mNydO-0uFE2fS8nBNE8f118tQAPo 3875
orbax/checkpoint/_src/path/format_utils_test.py sha256=sZc9BxoW57XKpyApzRVX6__bhXCGrY0DhI2udrzbwyM 4346
orbax/checkpoint/_src/path/step.py sha256=6caMRSPmBRsEgo3wZdX1rhGf177Kh7GU8AYylJJE1Wk 25180
orbax/checkpoint/_src/path/step_test.py sha256=6MHZoZx7PMdLDEIq5r-mi45gAaE2qWmYqN5AGYAWy0E 18279
orbax/checkpoint/_src/path/utils.py sha256=nkULJniV52MWQ_Y7fxg6IX3BMFSEdcCYi6OkIz7HukU 1411
orbax/checkpoint/_src/path/snapshot/BUILD sha256=-BXhMEguJ_KyBKXLte1KDrpfzhtTSiuyRDmFS6MELFY 333
orbax/checkpoint/_src/path/snapshot/snapshot.py sha256=EU2x6_OQB1wXB-pa6hy-Rnaisa_9DRB3G5fIcJ_3BiU 2271
orbax/checkpoint/_src/path/snapshot/snapshot_test.py sha256=kxUlzTfpTPVUI_z6nTOVEex4WMEt62XJwfT_wu-iE4g 1898
orbax/checkpoint/_src/serialization/BUILD sha256=8O3fsOTZHgVQnKGzmkfoo43PBIBJXHyKu8vMFVlIJR4 3246
orbax/checkpoint/_src/serialization/replica_slices.py sha256=5Lv10vnIBKgjuQHGm_IVlQ63IbRrmX5WzeCoP3HQ544 16083
orbax/checkpoint/_src/serialization/replica_slices_test.py sha256=-u-3x8iXV7GhSH-bfNiKpFc44pxLV1dzCIhgiV9zF7Q 5345
orbax/checkpoint/_src/serialization/serialization.py sha256=98BjivlvAbjhFup-SCjY4aXieneC39bkkdxKJMSppZc 20257
orbax/checkpoint/_src/serialization/serialization_test.py sha256=fE7Vb5O-vsLHxd-noiVrnmlWdlewWH6zpG3uLCbYHAE 22448
orbax/checkpoint/_src/serialization/tensorstore_utils.py sha256=vteCkQTe8th2g17cYAz1bjbbmlHNNGilOW-xs2Cg3Io 12866
orbax/checkpoint/_src/serialization/tensorstore_utils_test.py sha256=E-jfZ0uw-KwjCtOuQdKQB9at_hT6KQ0V6oj9K__7uWI 19695
orbax/checkpoint/_src/serialization/type_handlers.py sha256=6p6N7Z2UxrEoEWc3Td8n_j8pb8BMwaGYmVTGqPBf4ok 60624
orbax/checkpoint/_src/serialization/types.py sha256=0b9EhurEnRQwZJPZC025PsypADY0MBX4WLo47NQ_1nc 12448
orbax/checkpoint/_src/testing/BUILD sha256=dYIMsImcWZMzKZTG_7dcDo9pfI3O_QNXDX3zTZPwJe4 311
orbax/checkpoint/_src/testing/test_tree_utils.py sha256=uDlxkNxa2_ygbQfQs4sg1qAYhh2T7qPuIpFhzBEq34o 41791
orbax/checkpoint/_src/tree/BUILD sha256=M5g13rIvTGX9W16CtJKKDLLAOgkKHgL3HQKYjT6PHgg 534
orbax/checkpoint/_src/tree/types.py sha256=94QKfu4jo8ddkhU-8mRHW3KhhoJLc7CA6CpXFb8LvdA 1189
orbax/checkpoint/_src/tree/utils.py sha256=7QuIHPl1Hpg4aDEj9rbBTBsEtdKBZ-2JkVGUTMI0fPU 9906
orbax/checkpoint/_src/tree/utils_test.py sha256=Vub6jcS5D2nlHgpGYCVl7oGxjlgQdoGp-xlFJUkWNAk 8956
orbax/checkpoint/experimental/README.md sha256=tic5ktV4K_mb2HryBsCy59ocRU6P0trw5Rti_J4_-yM 115
orbax/checkpoint/experimental/__init__.py sha256=4Pb4JEUzytOA9FMwntBJ8un2Fmf1ARNNgHbC3tKnYqo 583
orbax/checkpoint/experimental/emergency/BUILD sha256=GODPkHcFnzFDzY6FaWmRQgRtEqtd_5C7RtwYHO_RpY0 3239
orbax/checkpoint/experimental/emergency/__init__.py sha256=4Pb4JEUzytOA9FMwntBJ8un2Fmf1ARNNgHbC3tKnYqo 583
orbax/checkpoint/experimental/emergency/checkpoint_manager.py sha256=t8xZWOWoMELn5kRWaz5kPtmlCBnw1Fg1Pi6qsTwnNgA 48467
orbax/checkpoint/experimental/emergency/integration_test_utils.py sha256=9ZA4amo8sO8oyRalGp8hT_ClzMDJpnm7tlhP03IA-VA 9238
orbax/checkpoint/experimental/emergency/local_checkpoint_data_debugging.py sha256=UB4DSnh9lFso1wlzQVen1TET_qXNLclDtaer7ZgPYR8 6765
orbax/checkpoint/experimental/emergency/mesh_consistency.py sha256=6-ZaSWtk3cf33bOrRLJ7qgbBOHLkr2uHPeMK5Zs5kxc 6743
orbax/checkpoint/experimental/emergency/multihost.py sha256=ksK5wnf6bHrUXCq4d5p4MlD0DZgrndt-Dt-osaPdAPM 6975
orbax/checkpoint/experimental/emergency/replicator_checkpoint_manager.py sha256=GwiN-FOOp0oksL0OSxkc2blyde5GQIqqRYYsh4tNp6Y 11898
orbax/checkpoint/experimental/emergency/test_utils.py sha256=_57drWi1cw12RNORt9m-QX8Vz65L4igJIZaJ4yTS7ro 46137
orbax/checkpoint/experimental/emergency/test_scripts/test_emergency_checkpointing.py sha256=zLjt_BBWnb0P-Y0z4jJ6GbrqRGI_6votNDhRJ_Y-lbk 11395
orbax/checkpoint/metadata/BUILD sha256=t-DAZqThxsV2IYBqF7A2oYSAGTAiOBbYAwYb89koSaE 687
orbax/checkpoint/metadata/__init__.py sha256=cuQTv4URIcZG73Pxr1RkGDkp16k7vxxce4kPT2nl7EU 1809
orbax/checkpoint/metadata/tree.py sha256=k317jm1t23hK0zZ4vKBNJ8XA0Zv2KKO6Fn6LyQKiPc8 908
orbax/checkpoint/metadata/value.py sha256=eY3GQ8pCaDCrL8GO0KX2aaNwi4htJEOWUtxfQfDFyKg 1047
orbax/checkpoint/multihost/BUILD sha256=FehknDArU9-W7HGwqwT17lz3a9L_UoFO1SCNFW9Ql0g 294
orbax/checkpoint/multihost/__init__.py sha256=j-hG9DloRUrX-Y2H2tRuLkWooSO9d6SOdqrfmMlJOW8 1559
orbax/checkpoint/multihost/multislice.py sha256=kCUAbVLVIxilkPlXKNeo6QHviv2YtGFaCUFOkngvQQQ 723
orbax/checkpoint/proto/__init__.py sha256=bdgIYbcjrXjdJPySS6o2gDD8F-ZoFPq6138AEtt9Ob0 757
orbax/checkpoint/proto/testing/BUILD sha256=vPtEiq2WtntcAXpAccBhQCwBpkyXxKkSOmUNOUuqBww 217
orbax/checkpoint/proto/testing/__init__.py sha256=bdgIYbcjrXjdJPySS6o2gDD8F-ZoFPq6138AEtt9Ob0 757
orbax/checkpoint/proto/testing/foo.proto sha256=AW4roV9hk7VSKx2cHXX36fCatv831362PomFj0f9Jzc 136
orbax/checkpoint/serialization/BUILD sha256=Y_Lv4EmRM3FaUMjL7rODfwX6gz11Beja2FTylXzs8hY 444
orbax/checkpoint/serialization/__init__.py sha256=hbkwmWDj3VkjVHqP465yYn5InUVosRpvUzJmNPJGjJY 971
orbax/checkpoint/serialization/tensorstore_utils.py sha256=vbqaZvRVb9CmrfDAQpZRQqGRR4nPmYRAlL-_KI_UHBQ 790
orbax/checkpoint/testing/BUILD sha256=fPAqFWEEUbMbRlOsCXtjfBqN1nphpNdDTNnij581pLA 445
orbax/checkpoint/testing/__init__.py sha256=CGd1WWZKizBiKu88buWxO8azSB2P7UgvBdMruT719o8 727
orbax/checkpoint/testing/aggregate_all.checkpoint/_METADATA sha256=6IKBbcEIYsG9X18eoB-c0xpN67gFgWEL-HFEeIaWbCE 637
orbax/checkpoint/testing/aggregate_all.checkpoint/checkpoint sha256=GOZGeUUMYeX7RMOKJmpHpuC0hgC68Sy0BzRXhVbfwys 454
orbax/checkpoint/testing/aggregate_some.checkpoint/_METADATA sha256=DzFiddtRCGmnbQjxFQISAYq3zpiTF3uAmAbZaIp2AfA 639
orbax/checkpoint/testing/aggregate_some.checkpoint/checkpoint sha256=O2elQvWDBTM92CvSGoaeOR3XM1ThfKSiOCxcl047Rk0 266
orbax/checkpoint/testing/aggregate_some.checkpoint/a/.zarray sha256=aa03md7vUuKGrrDCkuL7rDJqywStCJ5INyQZ94kidi4 164
orbax/checkpoint/testing/aggregate_some.checkpoint/a/0 sha256=lC75wFvyx-n7I4q77L_iBiRRjQNsoRhuSiCrZP-qjQk 38
orbax/checkpoint/testing/aggregate_some.checkpoint/b/.zarray sha256=vYUOuboOS6tsIPNgFqkym0ScpO8rYgtGQ9tUOdWH1Jk 166
orbax/checkpoint/testing/aggregate_some.checkpoint/b/0 sha256=3g-3oXsKM95BfwUGuw_Il9p-FmVsO-Q-F2itw2u8ItQ 55
orbax_checkpoint-0.11.1.dist-info/LICENSE sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0 11357
orbax_checkpoint-0.11.1.dist-info/WHEEL sha256=CpUCUxeHQbRN5UGRQHYRJorO5Af-Qy_fHMctcQ8DSGI 82
orbax_checkpoint-0.11.1.dist-info/METADATA sha256=6hIxsxaQXS_eyeR3LdDqSjlivkrLUSDzD5GipGL9K-I 1864
orbax_checkpoint-0.11.1.dist-info/RECORD