orbax/checkpoint/__init__.py,sha256=0HowmYKh-a-bvZmNtONWXvwHYc8Jr3OeMQn4fQKQpAE,4273
orbax/checkpoint/abstract_checkpoint_manager.py,sha256=aQpGrzp5s7lFRUGTf5KzOQ5fbkH1SUafuwrvtkmPJ5Y,5223
orbax/checkpoint/aggregate_handlers.py,sha256=aSzV4vGFcaFoGTcC7l3irUW828I7aMv97Mh0XPo7Jqg,3409
orbax/checkpoint/args.py,sha256=AZeZHgoX6KH0bKEBGE1PMmaELRfmEl96xwg2s5HNBFc,3116
orbax/checkpoint/arrays.py,sha256=DMR-o2bV6pYtRqY-o2dJdDh5oaE1ITWl7ZDK3cH_ea4,958
orbax/checkpoint/checkpoint_args.py,sha256=zIMPC7AGZpIc_9jtA6tu2e3AuiUkpo5cjf1Kws-LRwc,5664
orbax/checkpoint/checkpoint_manager.py,sha256=ziqcTZTtnAMRPbUmtXfXCfCsWo5z-LKu3jP1oCVpHbo,84367
orbax/checkpoint/checkpoint_managers.py,sha256=H0BC1OVhQOFm9RwZ5S2adW4vGB-YgpfhDj0l2fhm3U8,1718
orbax/checkpoint/checkpoint_utils.py,sha256=s2CXwFl3KUi7v0LBtkljHCFklXNPRnfmQKfnD4qGE1o,22288
orbax/checkpoint/checkpointers.py,sha256=bEtxBGJCg3O3yhjaC9E7vLW6v4mDYEs47-BC5Ux4_w8,1151
orbax/checkpoint/conftest.py,sha256=Q0fA9cTKiPfBhFwJTpw4j4x1Tb8Nsi6K3EcoTOZgoHw,740
orbax/checkpoint/future.py,sha256=EIJxYfa3fOzpxem0PmPTNzi8y0ntTWX3um5_jKhFMks,950
orbax/checkpoint/handlers.py,sha256=4OD2gWF8d2HqjiRajyDMxgh6TJHY3WQt65U1yOoXZjY,3764
orbax/checkpoint/msgpack_utils.py,sha256=RhvBIS9U05kk7rurHc4dPRyIcT4OzlHnaRDywPcZUD4,7673
orbax/checkpoint/options.py,sha256=uO4VkCCiS1XKQb73iPXeLjuLGDbxPdRRLmxZQnp5ulg,2664
orbax/checkpoint/path.py,sha256=Z4UUgSiayj0RIshr8ECHOyDPsoJZ_5ODP5o0C6A4-W0,989
orbax/checkpoint/pathways.py,sha256=3jYnxHSccGpGEvtmD5plkemxvxuLB2EoOSXC-1YBrCI,933
orbax/checkpoint/test_utils.py,sha256=-leonSx8dmbpW_DNYgcvEIJaA8Z32QTqkLqH5lThb48,26726
orbax/checkpoint/transform_utils.py,sha256=kpczafmH83r4RooWZ3BXkcjb2_RwbqOLSGnrdd4VS1M,12765
orbax/checkpoint/tree.py,sha256=nPQMD6YLVg7JJGjNf2cXAF0BS0NdKtW3OAkMJzkM5eQ,1141
orbax/checkpoint/type_handlers.py,sha256=gqIe2u8BTPr6y_SrxXhWLml4Vp_T6FzgvJ0oZ4zkQXQ,3374
orbax/checkpoint/utils.py,sha256=nNGf43AtLRYcVoDHdp4rIeadWJMHwTlGXsNigPkSHoo,6745
orbax/checkpoint/version.py,sha256=cUYtA4ArRX8nOTbjYHRr281XkGpBC-o0sK53vxfdeSE,1059
orbax/checkpoint/_src/README.md,sha256=y1gWZ2_EHkGviJfTn9tEhhDM-BkKAXbhd1L9Qh7YXWU,300
orbax/checkpoint/_src/asyncio_utils.py,sha256=6140p3sTZFZW2rtBI1dP2bK2z6_CdNT3li_mqq-952U,1136
orbax/checkpoint/_src/composite.py,sha256=2q18lC0YlWjy4BsURGdvaQdWMh27-qQP5bD00sB0JuQ,3286
orbax/checkpoint/_src/threading.py,sha256=WqQkdlI4ZgtebuhyOngp7flPLsc-BlwKSiVRZ23ZwQ8,3828
orbax/checkpoint/_src/arrays/abstract_arrays.py,sha256=o84BwdlHhptVEXr_H2Bzb52qibAyvcIuyoj0CIfs6AM,3686
orbax/checkpoint/_src/arrays/fragments.py,sha256=REnvk3_fI8gnwWMRiNXCGAEOMG1aVbPI2Rg1tyJ5qWQ,17454
orbax/checkpoint/_src/arrays/numpy_utils.py,sha256=UC0i5CxfY2vOMm_oX40i2l5Xu4RWrsrMannkqcZgadU,5172
orbax/checkpoint/_src/arrays/sharding.py,sha256=qY56dWDD_U_E_YvZonuTsERIw4Xbpli3mhAemQg4Qsc,4981
orbax/checkpoint/_src/arrays/subchunking.py,sha256=6IzaJkzwoLKnpsFN4ShpeLmnI7szmcrE7wn7x41SYGU,10388
orbax/checkpoint/_src/arrays/types.py,sha256=b6JLB2kr0uscCDHqT4AshHTVVN3SXeinwla5v2Ba9O8,1518
orbax/checkpoint/_src/checkpoint_managers/policy_checkpoint_info.py,sha256=4b15lziRG40BgsVoFy-BbrRHhKW2aBA9xCdRJoXBN20,845
orbax/checkpoint/_src/checkpoint_managers/preservation_policy.py,sha256=F6tuL_O0D-DB_QuSeDOGWb0I4c61wK00LXClW6myRmA,8529
orbax/checkpoint/_src/checkpoint_managers/save_decision_policy.py,sha256=gltR7hdinl3S3_lsnzqvr5ZTPn76LEjn1vAeBCKlDuA,6617
orbax/checkpoint/_src/checkpointers/abstract_checkpointer.py,sha256=zTTBzlL3xijEyKz_FpApq_SAQOyUaPCO7U3ZH02Ku3c,3028
orbax/checkpoint/_src/checkpointers/async_checkpointer.py,sha256=PxL0m1fqwR1tCFT1VdQRDXj3XeD1DAWUdw5ZF3emjHA,20293
orbax/checkpoint/_src/checkpointers/checkpointer.py,sha256=eRDMbouC67tqQlT5VlZoMFQfRG3Uwa2-7wSCOCn7rKc,15473
orbax/checkpoint/_src/checkpointers/checkpointer_test_utils.py,sha256=5n1fz0sga7j7PTC5S4MXHrXDgrnfhzl4LhiJwBC4rpM,30407
orbax/checkpoint/_src/checkpointers/pytree_checkpointer.py,sha256=IKYdamVBsPZm_eqeDk2oXF61POIprbedF1-nO26cYcE,1545
orbax/checkpoint/_src/checkpointers/standard_checkpointer.py,sha256=XAlAN7axJLRjJryKlwe-fj9CRz6e9k8qj-waF15Jx3s,5713
orbax/checkpoint/_src/futures/future.py,sha256=0euq4dxj6sL4k2lyPftyW46cUdHjuwVN_sqH48zKhRQ,15992
orbax/checkpoint/_src/futures/signaling_client.py,sha256=Rn5vEBdWAvrA7dd2YOOjL05sZXqIsRPyDJN0vEqoI1I,11034
orbax/checkpoint/_src/futures/synchronization.py,sha256=DeDJiXJD6aiR5NrOHYKWj83aHwXdhl2t_8r1kyPhDzw,4688
orbax/checkpoint/_src/handlers/array_checkpoint_handler.py,sha256=70k_83vDs36qDVRJvCa_C4e1GDTI607Mqn6_eRIrS04,7069
orbax/checkpoint/_src/handlers/async_checkpoint_handler.py,sha256=rpcolpOQZrRRH2g2y3FvdZLlaaIK25dfQZ7CmBorp1c,1562
orbax/checkpoint/_src/handlers/base_pytree_checkpoint_handler.py,sha256=UXNjAE75RuL0bFpo26-4B0BUjHjrG3J1ocU_Xjm090U,52253
orbax/checkpoint/_src/handlers/checkpoint_handler.py,sha256=B8C7PlORmKkjXzPGYXJgfd9nR1bnQN1IPPQ5g_wSQw0,2513
orbax/checkpoint/_src/handlers/composite_checkpoint_handler.py,sha256=ZmtWwKqkck3MGMX0hOAGd3Kalz3kvJfSkWSBnCGlYBQ,40090
orbax/checkpoint/_src/handlers/handler_registration.py,sha256=GypMn7LEwrloZn8vnIzFSZlXMuB0FLHsIkGsbjqc-NM,9371
orbax/checkpoint/_src/handlers/handler_type_registry.py,sha256=d3MpCxawqMgcvLraQrF7s7uYZdU-AgOV2Wba6_dsF4M,4100
orbax/checkpoint/_src/handlers/json_checkpoint_handler.py,sha256=J_ZmEJfrc6pmOTeikOhS0BO0J8bAoXN-i3njoIlFOjQ,4399
orbax/checkpoint/_src/handlers/proto_checkpoint_handler.py,sha256=zmeb7i6Bz-xdiDWfcQQ3Lt_ly_vXibrFR5fjKudMkh0,4116
orbax/checkpoint/_src/handlers/pytree_checkpoint_handler.py,sha256=-b1UysMBFIMGOhpquEby5oPoPLYF1iLBHloBEFh8pKg,47420
orbax/checkpoint/_src/handlers/random_key_checkpoint_handler.py,sha256=n0SB8RnvWBN_t0ro4pOAuAZIUH2jNG1tlWf-PXFq8_Y,9354
orbax/checkpoint/_src/handlers/standard_checkpoint_handler.py,sha256=95yhkFj9m10wR8JBLBiWE3N4Fp32epqArBNtF6O4c0U,13580
orbax/checkpoint/_src/handlers/standard_checkpoint_handler_test_utils.py,sha256=4BECRdoMpliUSrj_WnC-YoHSI0Lda_jjDf-pdPuLvLc,1204
orbax/checkpoint/_src/logging/abstract_logger.py,sha256=HTblqyIg6I5mPSz0P5FXn6kg7bCr_zrY16rysIQdRPE,1068
orbax/checkpoint/_src/logging/cloud_logger.py,sha256=Mp1GbzkpOP6d32FioPTG2pgilgaO0JVQhLv18KQJdyo,2584
orbax/checkpoint/_src/logging/composite_logger.py,sha256=D1FjNJBv2WIJuTsUdEk8SjKKxQjFs7CAWovA1fTGEwQ,1834
orbax/checkpoint/_src/logging/event_tracking.py,sha256=EWjDlj_IJc97R5BJl7z9LYgOw5IDvJFuJaIPd0VxHfg,2458
orbax/checkpoint/_src/logging/standard_logger.py,sha256=77tkF3rCaewYqgO4me7E-Gbk2elnaZIINpoTouW3hcg,1214
orbax/checkpoint/_src/logging/step_statistics.py,sha256=t0JLe99YmQJEQcc53C_ytDKvJIIPTpsqCsM-aqkGDs0,5530
orbax/checkpoint/_src/metadata/array_metadata.py,sha256=H5YcS-oLbPMdW5O-ZZcvlXRahQ0oJ4ehAFKCp9jvixQ,1677
orbax/checkpoint/_src/metadata/array_metadata_store.py,sha256=4j_xkdV8iNzAkmwytUGIKLurw0iLIPg_IHkMexha-jw,20979
orbax/checkpoint/_src/metadata/checkpoint.py,sha256=hPQ2A_wwimplTXoh-XMS9hG2vK2e__xgYD-A2RMtGSs,15826
orbax/checkpoint/_src/metadata/checkpoint_info.py,sha256=6hfS4fdtfCXCDHO_PDQ5h0w97PedsiPZ2hVoUZGDUWc,4428
orbax/checkpoint/_src/metadata/empty_values.py,sha256=YACcqn1FR6qSZFFfCRhPBPZ8qFsJNAPSVTtPdhMGyXA,4200
orbax/checkpoint/_src/metadata/metadata_serialization_utils.py,sha256=JjlTbhn-SdYDCMViOWIi8ZHY_zyq6G2722N5_VaKUn4,5184
orbax/checkpoint/_src/metadata/pytree_metadata_options.py,sha256=oyeaoTY7vdWqdo6_zwpNMYmGN4ev8ivde3N0UhgNAIM,1243
orbax/checkpoint/_src/metadata/root_metadata_serialization.py,sha256=Qi605Q2drjGk1TtWYrhKPFbkz3JGU6ghiZ4f27QwAXM,3449
orbax/checkpoint/_src/metadata/sharding.py,sha256=F5qbYx1ntRGXopWrT75q3oPZz8mTx0LDdZ5L7ZWTCOY,11648
orbax/checkpoint/_src/metadata/step_metadata_serialization.py,sha256=w1_AuKZgrIvVZt-3MxPCEkZM49LEsmUoJxF4m_SE9g8,8802
orbax/checkpoint/_src/metadata/tree.py,sha256=M3s7BxqsNnD1AcCHOYrLpjY7CVCWeks40OAvoP0YWO4,33489
orbax/checkpoint/_src/metadata/tree_rich_types.py,sha256=AAj3HQcRX4Rib6bHRRjv91iLUUceYQG7sh2I-jrL7Xs,7694
orbax/checkpoint/_src/metadata/value.py,sha256=pZdTvrHsLXojNtM1UOnUaG8ugqqASWRVXRYS2vHbQdw,3755
orbax/checkpoint/_src/metadata/value_metadata_entry.py,sha256=DvzcImxy0e6b2EV04CKmHijYtX600LHU0gi0vUW7PaU,2989
orbax/checkpoint/_src/multihost/dispatchers.py,sha256=3XQeJ34fDM0NN1xZiODTtpkfd81FEiBxu2HdlkFJXGs,12254
orbax/checkpoint/_src/multihost/multihost.py,sha256=cKQvuu34Xizv5zTmPoEZIqQV-9lRz5c5zqI-EMMYi5c,15712
orbax/checkpoint/_src/multihost/multislice.py,sha256=dNMGmYUO_MK2ncP6N8GcMWGstWEsARSfuOERhxV5Rgg,13731
orbax/checkpoint/_src/multihost/pathways.py,sha256=OoP3Vud_iyjFtMK9ZwdqlvUe5Bz56247NlGnIB7xHoQ,1347
orbax/checkpoint/_src/path/__init__.py,sha256=YnLhvIkalIQMcPQTpG5oWjYWzkTSqFOgTbJmpNnpFQ4,649
orbax/checkpoint/_src/path/async_path.py,sha256=YsIWwidi8ZP9tYu5CDckfPd0K1HBrJY7-vJHLgMnoio,2937
orbax/checkpoint/_src/path/atomicity.py,sha256=5SivnLLQ-uoVDOjHq5w_BsLjpTNPpCiOT7AyvdlPTto,26329
orbax/checkpoint/_src/path/atomicity_defaults.py,sha256=gbnIqxMAD5HBvA_E32_dT54SJLtJJ9f_5tc0QOCLjsY,1680
orbax/checkpoint/_src/path/atomicity_types.py,sha256=NG444wZeqADY4dgdHDUJFeOSJez9jvm7Bkx36lSeDWM,3487
orbax/checkpoint/_src/path/deleter.py,sha256=KyQ-KinN_snW_QdhcEpWcJTurP5pdf7clxL4JelM3O0,12046
orbax/checkpoint/_src/path/format_utils.py,sha256=PliCR75cC6S5jTtZYPFv__952A9RAlJ-8a_a5aTBNhU,4007
orbax/checkpoint/_src/path/gcs_utils.py,sha256=3uOS-h4Cc8TDq9Gn4KFr0qEzM8Y-_JretYklkaQqFkk,1991
orbax/checkpoint/_src/path/step.py,sha256=rnIRPTthXpdO46FldHhsOeQGATIjLrZGzgph2MTd3NU,28456
orbax/checkpoint/_src/path/temporary_paths.py,sha256=RCbh9UTncOU-9QqGkIOBSAFkyQBm9jCoSqj6RDPlNA0,4726
orbax/checkpoint/_src/path/utils.py,sha256=DZaDxEC0AcMVb8O8PBcCJJ9sPslhq6BZ55oFtZHzVsk,2498
orbax/checkpoint/_src/path/snapshot/snapshot.py,sha256=h6cuctWAUwvC-CEyIZj_-bAaI5S1HYMZeonJZmHM6VU,3490
orbax/checkpoint/_src/serialization/jax_array_handlers.py,sha256=gbrf-Rs7A-BqXrW4s385xu6OyKqvSTyNmlRRV9PrIoA,52839
orbax/checkpoint/_src/serialization/jax_array_restore_args.py,sha256=Flq1_pcgXuCW0N_-YUZivxN7sP_cSTndubXCx-zEeRY,4016
orbax/checkpoint/_src/serialization/limits.py,sha256=OQNB2MWSKNFbLPFO0ITnRwFchr9j8dfvkSYGgRNXNcE,3399
orbax/checkpoint/_src/serialization/ocdbt_utils.py,sha256=9R5_dFs9ufnP0x485EnUCSUQ2W-DHS-G9_h9EFDUJZs,6967
orbax/checkpoint/_src/serialization/pathways_handler_registry.py,sha256=P9Vz7s5LP7EF5eBE7HTxjZJOmCAh1mu3qoVuy7rgfxE,4352
orbax/checkpoint/_src/serialization/pathways_types.py,sha256=X_RcVKZbbXIfgRiDF8M7yIEFdIyQnx0uRfu9-TKzmto,1590
orbax/checkpoint/_src/serialization/replica_slices.py,sha256=G5SzibNs8YE1dZC-3kro0GlXzqlDYRIvurB7IeAr7mc,17477
orbax/checkpoint/_src/serialization/serialization.py,sha256=BSg3-0DB1uDqCwejApJjGDEu_bYOHk9jZ-7UZdPtuJ4,17576
orbax/checkpoint/_src/serialization/tensorstore_utils.py,sha256=luQuBiA2h9lZZbPqWUwHvc1T5pi-FaqipR-47OUDqtM,26499
orbax/checkpoint/_src/serialization/type_handler_registry.py,sha256=i4myovxF4jcjnU7sWhVboqC24FNqYIPMMTkV07e82Zk,8060
orbax/checkpoint/_src/serialization/type_handlers.py,sha256=jJBlvYCrewXqrUFeKwnNHp5CceRW5PKjZW5mZarBNj4,14580
orbax/checkpoint/_src/serialization/types.py,sha256=C2JR6HDI4fcQPWIstSiJnK9xVyaqKgxLaydI8lVNRdE,13888
orbax/checkpoint/_src/serialization/worker_memory_utils.py,sha256=KJdWftfswCEnh7bBEPHLTJLYqFk4yILo6Gs1wzc5jEU,7833
orbax/checkpoint/_src/testing/multiprocess_benchmark_configs.txt,sha256=etBiEOs5jc6_4X0Y4R2X6AxMaYHHeaTyPo0eqLsWdnA,204
orbax/checkpoint/_src/testing/multiprocess_tests.txt,sha256=N3WEgSxuIUd64ix8Fsa8qCMTLWLXXgCb5Ewiq0ltbfE,64
orbax/checkpoint/_src/testing/test_tree_utils.py,sha256=rYnibwOnqOYEi4o8Xs6su0GNf9gL4pHXcpPf3fgERfk,41791
orbax/checkpoint/_src/testing/benchmarks/array_handler_benchmark.py,sha256=mRz0qsq3N8kq0DUdNQwM54cSimzk90OtavnKE8GZaS8,8787
orbax/checkpoint/_src/testing/benchmarks/checkpoint_manager_benchmark.py,sha256=O8vI2MMI0IozA6pNNnf_Trslyy9dbrursmUyfhtNEnw,4740
orbax/checkpoint/_src/testing/benchmarks/checkpoint_manager_perf_benchmark.py,sha256=NSnbkikZqxwpc5gutV_MNASI6dbKA_np9xpgtBLBiaU,5271
orbax/checkpoint/_src/testing/benchmarks/checkpoint_policy_benchmark.py,sha256=x3e2gQcMBhVrf_EvIkJoObPiY__UMHVJ6xm24ta9gmM,10107
orbax/checkpoint/_src/testing/benchmarks/emergency_checkpoint_manager_benchmark.py,sha256=ofuFBWnu24oO09jek-CAA9JJ4RP4aUakWkxGb1VIAEw,10546
orbax/checkpoint/_src/testing/benchmarks/multihost_dispatchers_benchmark.py,sha256=7TH81Gsg43W9iB9deygoqH81HHE9cM73ki1nkikTQy0,6197
orbax/checkpoint/_src/testing/benchmarks/multislice_broadcast_benchmark.py,sha256=E9T5oVb135V86uJrFD9zWd6ZoBjCKN_sU3rc0yyjJu4,5209
orbax/checkpoint/_src/testing/benchmarks/pytree_checkpoint_benchmark.py,sha256=-wfGs4-3gxtWvcxoxQ3kLRfx5_6EQqzQQglyDrI0XFE,7404
orbax/checkpoint/_src/testing/benchmarks/run_benchmarks.py,sha256=R1DDR9l-ML2SL8mop2xlkrJzJ_1wof8N8D_Fv0PBXro,5431
orbax/checkpoint/_src/testing/benchmarks/single_replica_benchmark.py,sha256=PYhgq6YegLxE1DJkmIe430HYutypl1dtuGuW0bdkalk,6702
orbax/checkpoint/_src/testing/benchmarks/configs/array_handler_benchmark.yaml,sha256=i3-P8mEfadJRBLbkfvpnvxfjPVMzT9UwCRRwEfw8Y1Q,742
orbax/checkpoint/_src/testing/benchmarks/configs/checkpoint_manager_benchmark.yaml,sha256=BnnzbsVY1A7rfIV8uCCwRbPBsc5qqSyoIdPbp9HO0jE,583
orbax/checkpoint/_src/testing/benchmarks/configs/checkpoint_policy_benchmark.yaml,sha256=aHD7ezQT6FNrsMiLEYsE9bcaejqgjC4j4eujt2lkxO8,1625
orbax/checkpoint/_src/testing/benchmarks/configs/emergency_checkpoint_manager_benchmark.yaml,sha256=jeK-oXeCp1MWjklkLtxcsEQQBv_D3vPSOewJlJZo-Fs,2067
orbax/checkpoint/_src/testing/benchmarks/configs/multihost_dispatchers_benchmark.yaml,sha256=h7_Mh66HuaquYPEfkKV-ZllY_ydyXrmVIz0cADkFwvA,614
orbax/checkpoint/_src/testing/benchmarks/configs/multislice_broadcast_benchmark.yaml,sha256=ZuB9uSzz--aTBYInGe9xClOx98VGHz2TFS6p5zXzFOY,1591
orbax/checkpoint/_src/testing/benchmarks/configs/pytree_checkpoint_benchmark.yaml,sha256=ur-PIplGhYnqr0fQbOwHYYYNtg00-10JCmABRQIB8Bs,1791
orbax/checkpoint/_src/testing/benchmarks/configs/single_replica_benchmark.yaml,sha256=jm0wXIkucVtQoj6tsQxDa8MNhGYp0MU77RZNuhRBQPw,3428
orbax/checkpoint/_src/testing/benchmarks/core/checkpoint_generation.py,sha256=aglRUUxXku8k1CUShve_m6Y-AW6fW-nfmpJKw7utQRw,5055
orbax/checkpoint/_src/testing/benchmarks/core/config_parsing.py,sha256=P0hFlVuH3fLkFpvcJGyLC2bUa-165LVS3AvgkfZIuCA,6600
orbax/checkpoint/_src/testing/benchmarks/core/configs.py,sha256=tQAs8XKmt8ArgY5qV2TWUH01IV63pLgPUdl-VBRoD_M,2786
orbax/checkpoint/_src/testing/benchmarks/core/core.py,sha256=SLvnpP5hayP8UEh_k9136F4-T75Srvtpco89sPVI7Kg,14775
orbax/checkpoint/_src/testing/benchmarks/core/device_mesh.py,sha256=8EbHNsNX3Rdzz1xQv-xgjSxFatI1cv7orRqstcycuFw,3289
orbax/checkpoint/_src/testing/benchmarks/core/directory_setup.py,sha256=3qN20CWf7Gy9bATD_UTKeXgcriTuPARjdcaKa--Sej8,1722
orbax/checkpoint/_src/testing/benchmarks/core/mesh_utils.py,sha256=Fg8JhGSGdmp-7Hm-cx5hDLvaSzruKVHn8HfnKAP4lL0,2704
orbax/checkpoint/_src/testing/benchmarks/core/metric.py,sha256=vfTN1D6RovKuZYMtFF1RGEVFGEw2G_KchChOyYkSktw,22765
orbax/checkpoint/_src/testing/benchmarks/core/pytree_utils.py,sha256=8NF5GCRn3VscYykoPAknuRIio7zujf6HDTU3Igev4eI,2795
orbax/checkpoint/_src/testing/benchmarks/tpu_vm/README.md,sha256=I0FHohWLcrgzMTAXZSZgKI7cel2qIcm8FB6iuXwQkak,12583
orbax/checkpoint/_src/testing/benchmarks/tpu_vm/launch_on_vm.sh,sha256=r-rdUiaIuKJBdAhZN4WgyvsWjYO7Mrh-90e8uyQo9PI,10348
orbax/checkpoint/_src/testing/benchmarks/tpu_vm/manage_tpu.sh,sha256=z-7x4JW0kx1nRuc0F7IWCJVMMJdM5IePbExj-fNnDj0,7177
orbax/checkpoint/_src/testing/benchmarks/tpu_vm/setup_tpu.sh,sha256=rvhUVXP1nMSyZ9qLewBo1ljkXR7zeFUEI47OEQFqmpk,4021
orbax/checkpoint/_src/testing/benchmarks/xpk/Dockerfile,sha256=0gsfnMJ5c7bJH0WQyN-jZN6bQGim_4GgEP20fwfQPW0,3724
orbax/checkpoint/_src/testing/benchmarks/xpk/README.md,sha256=o-SQvzFSC5wfNHdDywt_j9C8U4pMqHw0PFGsdHwg9hs,16246
orbax/checkpoint/_src/testing/benchmarks/xpk/build_image.sh,sha256=trFdm5HDHTQXSd9ypGoYAVrr8XC4gSbjSlxNQjkw1YY,3774
orbax/checkpoint/_src/testing/benchmarks/xpk/launch_xpk.py,sha256=7XZr0Uym2vVkR0RRyaSWbwLWUfFT1diDe8j1OQYu2Kk,24986
orbax/checkpoint/_src/tree/parts_of.py,sha256=k36N9xTCV06cUjqD7h9GoSuOUw0KM3cV55CPjba9n8o,15101
orbax/checkpoint/_src/tree/structure_utils.py,sha256=obGcNGd4czQeNYhhdx977G8PO0mdAMhQFgwv-T-A2ow,21336
orbax/checkpoint/_src/tree/types.py,sha256=dDEsX8MNQOjAPUih8xuTt5lXI7uejkgDROULKwPPhxE,1189
orbax/checkpoint/_src/tree/utils.py,sha256=PCP8Ggg_aOLolL0IdyXEjluiM3Lw6ZW3L1BOr_6qPog,14720
orbax/checkpoint/experimental/README.md,sha256=tic5ktV4K_mb2HryBsCy59ocRU6P0trw5Rti_J4_-yM,115
orbax/checkpoint/experimental/__init__.py,sha256=qlEPFvPER9W0k5uJiqwZ-fvTrzsSr6i_c2J79TPx4t0,583
orbax/checkpoint/experimental/emergency/__init__.py,sha256=qlEPFvPER9W0k5uJiqwZ-fvTrzsSr6i_c2J79TPx4t0,583
orbax/checkpoint/experimental/emergency/checkpoint_manager.py,sha256=qIcmvkvJRFDiH1Wgp5lcOeJw2O3BUUoEdWFzFuHzoLY,55984
orbax/checkpoint/experimental/emergency/integration_test_utils.py,sha256=xbj9aJ1Vgl2Y-zzWKn_a4NevzORUkKeLulDbGep_wq8,9448
orbax/checkpoint/experimental/emergency/local_checkpoint_data_debugging.py,sha256=xFpGDsxAPmJL3zTY7xttZ9KV2zL_6mi9bXz5thIRvyA,6816
orbax/checkpoint/experimental/emergency/mesh_consistency.py,sha256=_cndY2L01wkhtkQTucJaglc3-mE9HCeASoj9_-Fj5jU,5989
orbax/checkpoint/experimental/emergency/multihost.py,sha256=R2GrkWEgsZ7JR6Z0M5rtUkHn3KRUGPRDekgRFOu-fAg,7878
orbax/checkpoint/experimental/emergency/path.py,sha256=yJKQmn1FnqyDvwkqmxW53D5rJM879dmpsYfNb3Iww4Y,4375
orbax/checkpoint/experimental/emergency/process_metadata_checkpoint_handler.py,sha256=ttAXfMzbB8RXb2I-atMdkJXlrMgGYaZ261L47BWHXWs,3651
orbax/checkpoint/experimental/emergency/replicator_checkpoint_manager.py,sha256=8O6pVltDnCKyXfz-csXnsRPAGifzroHpPjayrMqeZbE,17280
orbax/checkpoint/experimental/emergency/multi_tier_checkpointing/__init__.py,sha256=qlEPFvPER9W0k5uJiqwZ-fvTrzsSr6i_c2J79TPx4t0,583
orbax/checkpoint/experimental/emergency/multi_tier_checkpointing/initialization.py,sha256=6WjYjW1WAhTBVwc2HTCtCel8nJd_YBHL7ytXKsSpMfg,8915
orbax/checkpoint/experimental/emergency/test_scripts/test_emergency_checkpointing.py,sha256=Hm8Z5GVuLwsfWGYrNRxvMglxJoyIv9qfGVlHQDecJ1Q,12765
orbax/checkpoint/experimental/emergency/test_utils/dataset_iterator_checkpoint_handler.py,sha256=fmwSK3gHNCcah1bwWuWWL-1DoY3BhQTB_Yqw4FJ1E5w,4059
orbax/checkpoint/experimental/emergency/test_utils/test_base.py,sha256=TzJnN6xHbbzApYrGiY_lZEyFpKx_de82Le1ycZlkoUM,58506
orbax/checkpoint/experimental/v1/__init__.py,sha256=_hwppUbCi_tMVI2yYAKxkS-zEi5AcTjhWH7glVmZ51Y,2196
orbax/checkpoint/experimental/v1/arrays.py,sha256=9yTcY5ehWqLsQPTVaaOzauXP_axGgGznUJR-_2G25k8,819
orbax/checkpoint/experimental/v1/errors.py,sha256=z3thr5IxM0tHMjsYt0MCZ4KNwxXudrwt7ewAl3rgUhE,782
orbax/checkpoint/experimental/v1/handlers.py,sha256=qKrACm0dbcbvBcTvfr0JgS4ETF-NR0myO1-u_riiwUI,1381
orbax/checkpoint/experimental/v1/partial.py,sha256=99dN_EdlcUOzkJ7SWi7OJw64SsYyUPZabMjPTq_uWaQ,828
orbax/checkpoint/experimental/v1/path.py,sha256=uBp9Qopwad4I0s_hhVmybsYQNqPZLxGGzGwTaio2gZg,841
orbax/checkpoint/experimental/v1/serialization.py,sha256=zJh9A4uOe0fOxLj1ouCLMTgx9e5xTzE62nG4TEFrtfQ,1047
orbax/checkpoint/experimental/v1/training.py,sha256=A5hj_xgh2tqumBK0jbGW6585MPm--g_TlPB2rnXJf_A,1162
orbax/checkpoint/experimental/v1/tree.py,sha256=fGqDV0ZhZovym_fY3DVrh4QS10IjzTCUhroCQdS6Jn0,815
orbax/checkpoint/experimental/v1/_src/arrays/abstract_arrays.py,sha256=PTOW0Gxqal72VQfKNPXz4R-hyzzjPyBOzs2stO8sL70,754
orbax/checkpoint/experimental/v1/_src/context/context.py,sha256=VRTV2z2CwlAGJIemq7i3MrBT91yGx5jPh6fevGqkOAE,6590
orbax/checkpoint/experimental/v1/_src/context/options.py,sha256=MNGIZOFaeaFQC1iobM2G7IVHP5xGcqVT8Wngj0tpuos,15152
orbax/checkpoint/experimental/v1/_src/handlers/compatibility.py,sha256=pEVNnUBYc_G2MXzc07feT6Q2nL_GjBKElbh_NnXCcU4,4687
orbax/checkpoint/experimental/v1/_src/handlers/composite_handler.py,sha256=wxGwegqGratw0DKM7ORZ2A1qdXqRnubnlgukQTGC2Eg,12029
orbax/checkpoint/experimental/v1/_src/handlers/global_registration.py,sha256=F64aRrTTfDReEm8dGlVqwkxRH9X4u66DU-t7co2nctg,2071
orbax/checkpoint/experimental/v1/_src/handlers/json_handler.py,sha256=t5C6BdH7lV6PKUM9VLnAjYPJUEsPIJ67ZmNmwPR_MRk,3579
orbax/checkpoint/experimental/v1/_src/handlers/proto_handler.py,sha256=WgqGHuX4mXEm0vuWRn4l64Eo5q4dsMkAUYGAxdNJZis,3290
orbax/checkpoint/experimental/v1/_src/handlers/pytree_handler.py,sha256=FgzdpFP4VsT0_Uor4cV4_E3awo2BXiW8bV7YZBUfUzo,15268
orbax/checkpoint/experimental/v1/_src/handlers/registration.py,sha256=MGL38ozyCbLc_c4k2Y7lAnqtsbWntW3bMQGAuEq-JBg,16216
orbax/checkpoint/experimental/v1/_src/handlers/stateful_checkpointable_handler.py,sha256=K5cyUEb9CUYd8XOkiXSSxz92S94UagyHa_GRpWkU_LM,2519
orbax/checkpoint/experimental/v1/_src/handlers/types.py,sha256=CiwwOWIcgkdC6Lq6hOKi5x3VgEW59MoDFI_gRkw2rNM,10986
orbax/checkpoint/experimental/v1/_src/layout/checkpoint_layout.py,sha256=lvCbXpgc8K8ghv3W9hxXe4PpFk8NXus5d_t4W8YnjtE,3392
orbax/checkpoint/experimental/v1/_src/layout/numpy_layout.py,sha256=br7PCJUKFBhRkrfninFS7gpgf4uM6LUWJirDaSq8JMY,7515
orbax/checkpoint/experimental/v1/_src/layout/orbax_layout.py,sha256=TQSijeDHSGi8e-rttJv1uKlIm7Hez315UR0EQxaDMyk,10976
orbax/checkpoint/experimental/v1/_src/layout/pytorch_layout.py,sha256=dkRFptvewqe1E6NotobUuI0ByIdR4Tpt_RbyPiFzqFg,14028
orbax/checkpoint/experimental/v1/_src/layout/registry.py,sha256=dB31G5MUpC6mngjDi6b5p1NgUv8aoKM9j0zhtA7Mc14,5694
orbax/checkpoint/experimental/v1/_src/layout/safetensors_layout.py,sha256=4HZeBSu7dgTu7qzt4hcCc0vFPK5P20Sbm06LnRKO74s,11227
orbax/checkpoint/experimental/v1/_src/loading/loading.py,sha256=r_LU4aYVRCzyOmLUmqPp6ET1ZnNiTgLLtU9m1X9K-lY,10662
orbax/checkpoint/experimental/v1/_src/loading/v0_compatibility.py,sha256=NZvzjMz1qnJhPUKLDqB3VGOOgCPVB-gqpinxs-_1dis,4308
orbax/checkpoint/experimental/v1/_src/loading/validation.py,sha256=ABFiFRxTRpbP19gkmnoCZ80XFBNosqObNHKI6pH_Wfw,1290
orbax/checkpoint/experimental/v1/_src/metadata/loading.py,sha256=L7z9UZTCGy0Chzlo92bexMsjJen0OJdZ-U3z8Ria64A,5897
orbax/checkpoint/experimental/v1/_src/metadata/serialization.py,sha256=a9xvMHmHmXHl6kaa_b3W0dW3qWgYBGjaiwJUmmbny8k,3438
orbax/checkpoint/experimental/v1/_src/metadata/types.py,sha256=a6XoW8wtOkbwghcq9VJc4fww5nNRMzpDhQTy_uHWD6s,4446
orbax/checkpoint/experimental/v1/_src/partial/path.py,sha256=vrXIyd4AbJ0GTzWRocKMfj7yNKxVma-4Hhln491_pc0,1238
orbax/checkpoint/experimental/v1/_src/partial/saving.py,sha256=KA0b4g3oWRaxvMOFMksKoHD-eIrFhBEuRcS8yhyDLEw,12114
orbax/checkpoint/experimental/v1/_src/path/async_utils.py,sha256=keY6b7IH7iqQ5BohYMm_DLWYAMGXPKEYsfa3Vza2XG0,8732
orbax/checkpoint/experimental/v1/_src/path/step.py,sha256=MaKhwndqnBSaHMKG-UtXiAnYVYEIaAIjH44w7X7DoNw,876
orbax/checkpoint/experimental/v1/_src/path/types.py,sha256=APODOtRY_I3T8gP_kJS7NrRJoLKGgYjSoOkPRKeR45I,2059
orbax/checkpoint/experimental/v1/_src/saving/execution.py,sha256=jhItAdyK5cNYvTbS5YEYQGCz4kyptwJEq4cAsjwpZ8w,14116
orbax/checkpoint/experimental/v1/_src/saving/path_utils.py,sha256=WL0bEMqePHRwpt_7RZ5T5TG1yNC8Tlr-y-71LwsTyTo,4217
orbax/checkpoint/experimental/v1/_src/saving/saving.py,sha256=Yy1pyxNa4y4dizD7S9zENrlf8o0HbnojrXsUTFRXCxY,11263
orbax/checkpoint/experimental/v1/_src/serialization/array_leaf_handler.py,sha256=cDQETrOS16V-pgKED1jppZlkvjDsGg9hsZi_1hj0Dx8,11049
orbax/checkpoint/experimental/v1/_src/serialization/compatibility.py,sha256=pyaQVYmMqSDeWpBgsLima9H_pfNitR_C2TddgzO-Vcs,15531
orbax/checkpoint/experimental/v1/_src/serialization/numpy_leaf_handler.py,sha256=px_ylek-Xb1HYh-0pjCskTkNAWzFCqA7W3xAIpIFPjE,8616
orbax/checkpoint/experimental/v1/_src/serialization/protocol_utils.py,sha256=T2jYvoAEMybBczDrFWh2ZxMfTcqw91jv_g17Qz1igIQ,2076
orbax/checkpoint/experimental/v1/_src/serialization/registry.py,sha256=N1uELRmjApBnQtAWtwcQLXwRtusCDqpXSL5r_UfbrOY,7344
orbax/checkpoint/experimental/v1/_src/serialization/scalar_leaf_handler.py,sha256=EkRRcTsBbVsDEcC0vBZUAljnMKNgHt89gSv3d0lIwvM,8285
orbax/checkpoint/experimental/v1/_src/serialization/string_leaf_handler.py,sha256=0Pm6NabSVvEAeaTOEHSSaVSeZXNxIerfaYThbtDVRDM,6583
orbax/checkpoint/experimental/v1/_src/serialization/types.py,sha256=ggA4RAfoKkPcJ7X4y1N2Gs70WI0mcLxqOR32NcHLP74,9576
orbax/checkpoint/experimental/v1/_src/synchronization/asyncio_utils.py,sha256=s3IRSxfOIIskTDdMw-NfS00XqiPdAy_yA_Bdi7QYi00,735
orbax/checkpoint/experimental/v1/_src/synchronization/multihost.py,sha256=MPZ3IM3bUKD6l8whP_My86FfH4MWIGnhinF6UZnD4jY,5125
orbax/checkpoint/experimental/v1/_src/synchronization/signaling_client.py,sha256=BfQNJVI-rf-PxCQs6gLkd53r6w8ovWlYQ_mvuV7Xp0s,3538
orbax/checkpoint/experimental/v1/_src/synchronization/synchronization.py,sha256=TvP9EC8wPxLuMXqF8ON_RKEDtlc3VXWxqvZTExcIcQ4,2870
orbax/checkpoint/experimental/v1/_src/synchronization/thread_utils.py,sha256=N_Y6nia5_Id-zQgEGIrixAwmp2xbz9hL1mCGXY8EdNc,1904
orbax/checkpoint/experimental/v1/_src/synchronization/types.py,sha256=g4X49qN5Ss4jSiJ0zk8Fn1siRImN7ixO6TWImTurgBo,793
orbax/checkpoint/experimental/v1/_src/testing/array_utils.py,sha256=Nw_2ATvCK1EPqDAqm3xAz3kAZJIqZbp6WNUE0crxFvU,3783
orbax/checkpoint/experimental/v1/_src/testing/handler_utils.py,sha256=MJKOVlGSIIkqsFU750jF827BH9IWyg71JeFIijNz7jA,10846
orbax/checkpoint/experimental/v1/_src/testing/path_utils.py,sha256=TxsHGafYqAURPGwmtxHYFK7xTqZpBJVrArIEyyeSvvI,1473
orbax/checkpoint/experimental/v1/_src/testing/save_load_test_base.py,sha256=1SExGvIpJS7MrCDqo_yuOpQHZyE7n0EiZcbWSqFd810,36334
orbax/checkpoint/experimental/v1/_src/testing/tree_utils.py,sha256=grbaXfGp9-Wo4taInKdP5_W1r5iNP3xWOGm9Y2Ch0Sk,2599
orbax/checkpoint/experimental/v1/_src/testing/v0v1_compatibility_save_load_test_base.py,sha256=k4j9sLjUSMXxdCd1K14uX0DDfG9TTvJjA__ARYhrecE,10506
orbax/checkpoint/experimental/v1/_src/training/checkpointer.py,sha256=IeeCuIoebz15sm59q8ev8tdg_X-vTh5T3xPeQWnjv8o,20566
orbax/checkpoint/experimental/v1/_src/training/checkpointer_test_base.py,sha256=iNX0VedeampEiQFIN3gO_-whQBki_-roSn2NR8qG0E0,25678
orbax/checkpoint/experimental/v1/_src/training/errors.py,sha256=oKXbRoTmzrZ_E_fPPaAW3A0gsjtkQe90acC7Uk56YLs,834
orbax/checkpoint/experimental/v1/_src/training/model_helpers.py,sha256=Mnq6i9mRFywJqryj1lySPUClrMb9gyA9QL_S5egh8tk,2409
orbax/checkpoint/experimental/v1/_src/training/preservation_policies.py,sha256=ttXF2-SYgsH8e14hz4lX3XTgEOwnblY6M0EpxmPK8y0,1704
orbax/checkpoint/experimental/v1/_src/training/save_decision_policies.py,sha256=7B-uipr84dWpRdHhLYvF5IsxRGQgAiP1j4IGrx1EdC8,1699
orbax/checkpoint/experimental/v1/_src/training/v0v1_compatibility_checkpointer_test_base.py,sha256=3zHPP7tL3iwdxkf9PZHtSdZPhUI0_jg8XuJ-j3N1FRY,8001
orbax/checkpoint/experimental/v1/_src/training/metadata/types.py,sha256=ig8KwDNy17yxPfHklAFVPd7FOpw8eo9BVYlqVIAVBrk,3526
orbax/checkpoint/experimental/v1/_src/tree/structure_utils.py,sha256=b6o9xvpFGVMfH6LwG6gneNE_qXSvg7Bau0NdfyOJ4ds,745
orbax/checkpoint/experimental/v1/_src/tree/types.py,sha256=toGurIxvizf43dmgWx-tjqh7SO8mnaBNUst3vIoHbQQ,1058
orbax/checkpoint/logging/__init__.py,sha256=A7IlLsHHGxMDwvkCJ2yqv2KLGL60aTgWHOyURGIzljs,1199
orbax/checkpoint/logging/step_statistics.py,sha256=rJwNze9aP7BGbKt0rvIvtxwr-TF62UOZ6Eos24O221g,881
orbax/checkpoint/metadata/__init__.py,sha256=139OefTou4Tp-77JIXiPqKR5si54YT4PveGMWQ9Dius,1745
orbax/checkpoint/metadata/tree.py,sha256=mkAx2uMCyd8rDndPmjQRSAJtvku6HzpHyojnw42slOA,908
orbax/checkpoint/metadata/value.py,sha256=nSru4FpRD0lBTBTzhyXYMibMwP930CHIqbcE6tV_a5Q,1047
orbax/checkpoint/multihost/__init__.py,sha256=nqC_FdI1SIBfZujdMI4j4Big0vAIwjoMWG-3ytJXntc,1559
orbax/checkpoint/multihost/multislice.py,sha256=qvUqxoX-i2kn_RVe4fVo3A014NfSuiS8CkCvQQZY8oU,723
orbax/checkpoint/serialization/__init__.py,sha256=rXredCQ0pgrTiL4f3etughqojlmwXEJFPLcBn24JDSU,971
orbax/checkpoint/serialization/tensorstore_utils.py,sha256=avp_ccWsLCENz8cAtqoB7aONWIcd4X9vplbqBt44SJ0,790
orbax/checkpoint/testing/__init__.py,sha256=m7o0SEYqQT25E2a53wPGUWFcCSvaQP06JGvNQo3sVzc,727
orbax/checkpoint/testing/local_path.py,sha256=GDW-aorX9i5MoE3jTCOkoFn2LE8JNQpGkF6M0RNzYFA,6696
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.32.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
orbax_checkpoint-0.11.32.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
orbax_checkpoint-0.11.32.dist-info/METADATA,sha256=S_cuyymkmPW-3IRb0GjdbZfH6KqZvVAw1B1m7Z-VOtc,2530
orbax_checkpoint-0.11.32.dist-info/RECORD,,
