{"library":"orbax-checkpoint","type":"library","category":null,"description":"Orbax Checkpoint provides a robust, asynchronous, and fault-tolerant checkpointing library primarily designed for JAX and Flax models. It allows saving and restoring large-scale machine learning model states and arbitrary data structures efficiently, supporting distributed environments and custom serialization. It is currently at version 0.11.34 and frequently updates to align with JAX/Flax developments.","language":"python","status":"active","version":"0.11.34","tags":["jax","flax","checkpointing","machine-learning","serialization"],"install":[{"cmd":"pip install orbax-checkpoint jax flax","imports":["import orbax.checkpoint as ocp","from orbax.checkpoint import CheckpointManager","from orbax.checkpoint import CheckpointManagerOptions","from orbax.checkpoint import StandardSave","from orbax.checkpoint import StandardRestore"]},{"cmd":"pip install orbax-checkpoint","imports":[]}],"homepage":null,"github":"http://github.com/google/orbax","docs":null,"changelog":null,"pypi":"https://pypi.org/project/orbax-checkpoint/","npm":null,"openapi_spec":null,"status_page":null,"smithery":null,"compatibility":{"summary":{"python_range":"3.10–3.9","success_rate":63,"avg_install_s":15.9,"avg_import_s":3.48,"wheel_type":"wheel"},"url":"https://checklist.day/v1/registry/orbax-checkpoint/compatibility"},"provenance":{"verified_status":"install_fail","verified_at":"Sun Jun 28","last_verified":"Sun Jun 28","next_check":"Sun Jul 05","install_tag":null}}