Wheelodex » Projects » orbax-checkpoint

orbax-checkpoint

View on PyPI — Reverse Dependencies (25)

0.10.3 orbax_checkpoint-0.10.3-py3-none-any.whl

Wheel Details

Project: orbax-checkpoint
Version: 0.10.3
Filename: orbax_checkpoint-0.10.3-py3-none-any.whl
Download: [link]
Size: 359825
MD5: cf65ec9c07d9e283efd57e6b826b7fd0
SHA256: df7fd5f327dfe9c477533f33c20076ae11ba6a15767c5117881b328dece14c7d
Uploaded: 2024-12-19 20:25:46 +0000

dist-info

METADATA · WHEEL · RECORD

METADATA

Metadata-Version: 2.3
Name: orbax-checkpoint
Version: 0.10.3
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/__init__.py sha256=045slIyb-_GODUhE2mDjsD07cr8Ig7Be2m8YOo_F_DE 3955
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=OyXvVZY1r1N97Qzd1DRFZfHRJqUuZ8pDjQsFCVK7-qA 71950
orbax/checkpoint/checkpoint_utils.py sha256=f4pUch8-LIMYOYlSBJbamX_QaABAvG6EuYTxveMB6T4 18531
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=DyzK8yY4hsQ_ZaBNFWKip4xEb40F6V8Ba77X4G1CiLY 3070
orbax/checkpoint/handlers.py sha256=Svz3keBxEa_UOub082bZkIFZQe2nDoVrwG_McfxNpRs 3665
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=VgvOEST82wW_eDxqNmZAcu_g7MvYcNyrqd1Jehp2QDo 2649
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=qf1rxc3Jij_1L__FXzjIr8YMIAXIkcDba3sX7YQo0bs 22383
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=COQG6nMw1QLm8DJUde7m-sOFhkqheyu0a2vhXzR9brs 984
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/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=pYZOodbwb4POBrPl2vRyeo7GEfCPU8C7HycNY1pVFOw 1459
orbax/checkpoint/_src/checkpointers/abstract_checkpointer.py sha256=kXtf7ARoaJPppaQFqVjvd7sB7WMR7ovOY3Q-c96PyZM 3015
orbax/checkpoint/_src/checkpointers/async_checkpointer.py sha256=jzgAZABPNL_b1L3xHTJQqek4SAsfwBEzjDO597cth94 16122
orbax/checkpoint/_src/checkpointers/checkpointer.py sha256=jcLCf3wm41c3h_hACBCtn1yuPVXM-M_-na2DYRiKxns 10016
orbax/checkpoint/_src/checkpointers/pytree_checkpointer.py sha256=f4pXIkNTVeE7mdDEKF5kgRDGORLwnGpStIzidLPNM2g 1451
orbax/checkpoint/_src/checkpointers/standard_checkpointer.py sha256=cEds3DDjJmF4Bpye_3t4UInyAhhVIiCtfMr539GBNOQ 5455
orbax/checkpoint/_src/handlers/array_checkpoint_handler.py sha256=6QLlLnB0iuUiKzlISF5vZ4qTjdyYL9uaYWX5Y16mqWg 7186
orbax/checkpoint/_src/handlers/async_checkpoint_handler.py sha256=AIB08lSVIW9NyRPIio0vJ2NnkupnaQaFwzLJQwXNCoA 1392
orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py sha256=iTK_bKr-M8fzlRFjtZTmWNa4DeNWN8bon6xLJ8MenqQ 31567
orbax/checkpoint/_src/handlers/checkpoint_handler.py sha256=ihHJ9TMsMIjfdUjKNFsilmVrhvFtcq8-awkxB2xg1Uc 2479
orbax/checkpoint/_src/handlers/composite_checkpoint_handler.py sha256=Vu85rsOFANQG9-TRvpoHiuuz8xRfzX9q5tYEzXQveOQ 35416
orbax/checkpoint/_src/handlers/composite_checkpoint_handler_test.py sha256=Olf9-EQxJChbF1SNE1Ub1JajTmg5G2MPv0tU6a41IrU 33130
orbax/checkpoint/_src/handlers/handler_registration.py sha256=kEvF8mMcpuyXzFdPP92Whap5cIOwUYuUJPgATt5tup0 8852
orbax/checkpoint/_src/handlers/handler_registration_test.py sha256=_t8f9dTr9pWlMyuOYiIRTZjAeqfoDBMMfy2i1lI4Y2g 6984
orbax/checkpoint/_src/handlers/handler_type_registry.py sha256=0ZLGXOe7V9hh5ff2pVpTIufVLDNs5RZhrJyzbtCchos 2379
orbax/checkpoint/_src/handlers/handler_type_registry_test.py sha256=f2rBQpP07RTWbuw_ZX3ag4SGmZOWyMRVtcKnhv-dnHM 5889
orbax/checkpoint/_src/handlers/json_checkpoint_handler.py sha256=o0tzM9a8RjS7RKqWxuU4LN-WpbLlg4jMTVB0atAKB5k 4455
orbax/checkpoint/_src/handlers/json_checkpoint_handler_test.py sha256=supNdKD5lYyRrzhLy8KqkrJOLORX-o8X28wyez5S7mc 3227
orbax/checkpoint/_src/handlers/proto_checkpoint_handler.py sha256=FgZVSP_f7Jm76jUE-Ktdef7odmKnPmdRHiAXXq6ybFI 4218
orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py sha256=rQmgO_NTUyLV68gdHYi5xgphZIUIt3Mv2TqJl0Z1YUQ 42096
orbax/checkpoint/_src/handlers/random_key_checkpoint_handler.py sha256=aWJq4uxe6NG7f8PwTsnmkYA1e-O9HZdL7b_OIkEuGg4 9101
orbax/checkpoint/_src/handlers/random_key_checkpoint_handler_test.py sha256=K608d11IiZtR5usllL5r4_uIDCsKiA-jmlMwkoXcOAA 6911
orbax/checkpoint/_src/handlers/standard_checkpoint_handler.py sha256=cB1PmlokubefEKafXuOkBQq70KHsltaWo6S6l0qWCUY 10431
orbax/checkpoint/_src/handlers/standard_checkpoint_handler_test_utils.py sha256=QA_0HY3mNsOLZbcgZ1g3mTtDM1xbRJZJ6xIijng38HM 14303
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/checkpoint.py sha256=KS24cNQH6u4atLBNxBzVtB-LwdYxuBbnhxfmu3jQfeQ 15692
orbax/checkpoint/_src/metadata/checkpoint_test.py sha256=I_Hlot2gPo6JxmA3NbVqg6CFgVgxlRGWI6AOS12-o1c 24191
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=UQW10TCPRIuhTiuD5fJYBrPCqlmXJc2u1LQn64bQCso 2510
orbax/checkpoint/_src/metadata/pytree_metadata_options.py sha256=UgNRGHZPVw5YO_NV5LQKml-_fWNZIKc4A1iG2f7Mcqo 1235
orbax/checkpoint/_src/metadata/root_metadata_serialization.py sha256=V5QD3TThIOYcAXdZNTQvIa7vIEJqZNb68Q6ELbEhwy8 2333
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=kTRqkQnwS6JGrhj2K-6xDaEZ_cdAnnM5akmIiE9_4-U 7393
orbax/checkpoint/_src/metadata/tree.py sha256=a0QN4g2TNAVXPGmKX-bEyj1spExpsPTxlXjiOnUXm_0 17561
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=7brFghC_BI3nYCRATFT_PeBx4MbjIRZvw_ZctQrBeQw 4725
orbax/checkpoint/_src/metadata/value.py sha256=rL5DIR3f6AKhl6kc8Z7ZyG6vSoFO0dC6q9-Sa-5zy1s 3632
orbax/checkpoint/_src/metadata/value_metadata_entry.py sha256=2GKWXO5fiC5Ul1fXzFcUVZBNCItS6kGPY-oJdX2Gb1s 2424
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=ZQ8K7wxwBH4gAsp6kJaGq9K80IoQ_4GIoFeAYNReJbA 267
orbax/checkpoint/_src/path/__init__.py sha256=9xDrETQE_Fq-P-NVacRNKcrVOEdoltsbm7FIRTl5lqc 649
orbax/checkpoint/_src/path/async_utils.py sha256=xnlgYSC5V2V1GFXHMyDFYMfL8vl36mO9RImE8DmttsU 1676
orbax/checkpoint/_src/path/atomicity.py sha256=6SPqahX1_-AqVRoPU50r4A5NXbGlhithv08-VI-Pg3E 16009
orbax/checkpoint/_src/path/atomicity_defaults.py sha256=b9m3DxYq5N5oNEqp2RG1P2kihqfpDII95IMBgrLf8Gk 1325
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=OSmQncd4ABWDGxw7JTWBUqVw2dhAH91XXGHI_QK6qn0 448
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/replica_slices.py sha256=AEiiS7A2RsBitTNSkkrI-Nv4dtQ0-0995sPsmpZo6yQ 13162
orbax/checkpoint/_src/serialization/replica_slices_test.py sha256=-u-3x8iXV7GhSH-bfNiKpFc44pxLV1dzCIhgiV9zF7Q 5345
orbax/checkpoint/_src/serialization/serialization.py sha256=HeFr9HKWe5hdbtBw5t3aP7J5a2yuh4sDG6GpwEmyYJQ 20256
orbax/checkpoint/_src/serialization/serialization_test.py sha256=urDh-tmWTKUO0Fop5rR48LSJY6zFP3XaWryIyULa3fw 22435
orbax/checkpoint/_src/serialization/tensorstore_utils.py sha256=vHNhg2KEi2AcW2fOnrlCIwCtsDnvffp_w-_oe782234 12989
orbax/checkpoint/_src/serialization/tensorstore_utils_test.py sha256=E-jfZ0uw-KwjCtOuQdKQB9at_hT6KQ0V6oj9K__7uWI 19695
orbax/checkpoint/_src/serialization/type_handlers.py sha256=dVBClfo6Bh1TIlUJhwVTpZBF2gI8LgoVbvlHWr1Ll98 59181
orbax/checkpoint/_src/serialization/types.py sha256=J_vDqTfTXQ0boRTOk_q5zZWP8lk643nQLQw35k3y9nA 12004
orbax/checkpoint/_src/testing/test_tree_utils.py sha256=6eNMCD9seG-t6_K1Iwt9R8docQDXP3014l7_ejvCfQs 41720
orbax/checkpoint/_src/tree/types.py sha256=4sQmw2uwzOgCIzH6VCUjwNnkJmLeEzvv0wMUqFia8jM 1043
orbax/checkpoint/_src/tree/utils.py sha256=qgDia5aYGBfdWLq2CmEY1XGtH-X32DBab3gxLG0eJjA 9740
orbax/checkpoint/_src/tree/utils_test.py sha256=Wqos74bgd9Gx9H8FgYIosWCQALDqn_hzX1OfFC49r9o 8317
orbax/checkpoint/experimental/README.md sha256=tic5ktV4K_mb2HryBsCy59ocRU6P0trw5Rti_J4_-yM 115
orbax/checkpoint/experimental/__init__.py sha256=4Pb4JEUzytOA9FMwntBJ8un2Fmf1ARNNgHbC3tKnYqo 583
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=rhhMewPor-P9z14vCbGXjpnyKhB6q31d5ev2mp-ve4A 6722
orbax/checkpoint/experimental/emergency/multihost.py sha256=3_BPncWb06550tPYaGJLgOSAXPAOL2y0P1aZ-evKiLc 6915
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/__init__.py sha256=cuQTv4URIcZG73Pxr1RkGDkp16k7vxxce4kPT2nl7EU 1809
orbax/checkpoint/metadata/tree.py sha256=k317jm1t23hK0zZ4vKBNJ8XA0Zv2KKO6Fn6LyQKiPc8 908
orbax/checkpoint/metadata/value.py sha256=eY3GQ8pCaDCrL8GO0KX2aaNwi4htJEOWUtxfQfDFyKg 1047
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/__init__.py sha256=bdgIYbcjrXjdJPySS6o2gDD8F-ZoFPq6138AEtt9Ob0 757
orbax/checkpoint/proto/testing/foo.proto sha256=AW4roV9hk7VSKx2cHXX36fCatv831362PomFj0f9Jzc 136
orbax/checkpoint/serialization/__init__.py sha256=hbkwmWDj3VkjVHqP465yYn5InUVosRpvUzJmNPJGjJY 971
orbax/checkpoint/serialization/tensorstore_utils.py sha256=vbqaZvRVb9CmrfDAQpZRQqGRR4nPmYRAlL-_KI_UHBQ 790
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.10.3.dist-info/LICENSE sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0 11357
orbax_checkpoint-0.10.3.dist-info/WHEEL sha256=CpUCUxeHQbRN5UGRQHYRJorO5Af-Qy_fHMctcQ8DSGI 82
orbax_checkpoint-0.10.3.dist-info/METADATA sha256=JXEV0EaZIJGG4GB5r2Cjh9NAtciAv-3-cl_WHGUnI5Y 1864
orbax_checkpoint-0.10.3.dist-info/RECORD — —