# Orbax Checkpointing
`pip install orbax-checkpoint` (latest PyPi release) OR
`pip install 'git+https://github.com/google/orbax/#subdirectory=checkpoint'` (from this repository, at HEAD)
`import orbax.checkpoint`
Orbax includes a checkpointing library oriented towards JAX users, supporting a
variety of different features required by different frameworks, including
asynchronous checkpointing, various types, and various storage formats.
We aim to provide a highly customizable and composable API which maximizes
flexibility for diverse use cases.
Raw data
{
"_id": null,
"home_page": null,
"name": "orbax-checkpoint",
"maintainer": null,
"docs_url": null,
"requires_python": ">=3.10",
"maintainer_email": null,
"keywords": "JAX machine learning, checkpoint, training",
"author": null,
"author_email": "Orbax Authors <orbax-dev@google.com>",
"download_url": "https://files.pythonhosted.org/packages/66/3a/60a08a16997b1e4e097d03b8b9135eec44c88cb8547822efe46d2e97eb09/orbax_checkpoint-0.11.1.tar.gz",
"platform": null,
"description": "# Orbax Checkpointing\n\n`pip install orbax-checkpoint` (latest PyPi release) OR\n\n`pip install 'git+https://github.com/google/orbax/#subdirectory=checkpoint'` (from this repository, at HEAD)\n\n`import orbax.checkpoint`\n\nOrbax includes a checkpointing library oriented towards JAX users, supporting a\nvariety of different features required by different frameworks, including\nasynchronous checkpointing, various types, and various storage formats.\nWe aim to provide a highly customizable and composable API which maximizes\nflexibility for diverse use cases.\n\n\n\n\n",
"bugtrack_url": null,
"license": null,
"summary": "Orbax Checkpoint",
"version": "0.11.1",
"project_urls": {
"homepage": "http://github.com/google/orbax",
"repository": "http://github.com/google/orbax"
},
"split_keywords": [
"jax machine learning",
" checkpoint",
" training"
],
"urls": [
{
"comment_text": null,
"digests": {
"blake2b_256": "000690f972344350d12ed77b6cf28f8e06215eff9d8ca4da95679e8d0369d697",
"md5": "607a6fb773f46e88d652bf0ff700a643",
"sha256": "9cfbe49331ffdef4aedfec337da683eb8fc65221dc43924cf6daea6c27d6dc06"
},
"downloads": -1,
"filename": "orbax_checkpoint-0.11.1-py3-none-any.whl",
"has_sig": false,
"md5_digest": "607a6fb773f46e88d652bf0ff700a643",
"packagetype": "bdist_wheel",
"python_version": "py3",
"requires_python": ">=3.10",
"size": 394885,
"upload_time": "2025-01-28T23:09:18",
"upload_time_iso_8601": "2025-01-28T23:09:18.688781Z",
"url": "https://files.pythonhosted.org/packages/00/06/90f972344350d12ed77b6cf28f8e06215eff9d8ca4da95679e8d0369d697/orbax_checkpoint-0.11.1-py3-none-any.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": null,
"digests": {
"blake2b_256": "663a60a08a16997b1e4e097d03b8b9135eec44c88cb8547822efe46d2e97eb09",
"md5": "b07b5fda648f9f058b0e5e362938d6fc",
"sha256": "e303f4f4441ba0367a14c692d6c73070bba954205de89df608f21823dd86eaee"
},
"downloads": -1,
"filename": "orbax_checkpoint-0.11.1.tar.gz",
"has_sig": false,
"md5_digest": "b07b5fda648f9f058b0e5e362938d6fc",
"packagetype": "sdist",
"python_version": "source",
"requires_python": ">=3.10",
"size": 277832,
"upload_time": "2025-01-28T23:09:21",
"upload_time_iso_8601": "2025-01-28T23:09:21.658753Z",
"url": "https://files.pythonhosted.org/packages/66/3a/60a08a16997b1e4e097d03b8b9135eec44c88cb8547822efe46d2e97eb09/orbax_checkpoint-0.11.1.tar.gz",
"yanked": false,
"yanked_reason": null
}
],
"upload_time": "2025-01-28 23:09:21",
"github": true,
"gitlab": false,
"bitbucket": false,
"codeberg": false,
"github_user": "google",
"github_project": "orbax",
"travis_ci": false,
"coveralls": false,
"github_actions": true,
"requirements": [
{
"name": "jax",
"specs": [
[
">=",
"0.4.34"
]
]
},
{
"name": "jaxlib",
"specs": [
[
">=",
"0.4.34"
]
]
},
{
"name": "portpicker",
"specs": [
[
"~=",
"1.6"
]
]
},
{
"name": "absl-py",
"specs": [
[
"==",
"1.*"
],
[
">=",
"1.0"
]
]
},
{
"name": "numpy",
"specs": [
[
">=",
"1.26.0"
]
]
},
{
"name": "orbax-checkpoint",
"specs": [
[
">=",
"0.9.0"
]
]
},
{
"name": "etils",
"specs": []
},
{
"name": "etils",
"specs": []
},
{
"name": "simplejson",
"specs": []
},
{
"name": "chex",
"specs": []
},
{
"name": "optax",
"specs": []
},
{
"name": "mock",
"specs": []
},
{
"name": "nest_asyncio",
"specs": []
},
{
"name": "tensorstore",
"specs": []
},
{
"name": "humanize",
"specs": []
},
{
"name": "flax",
"specs": []
},
{
"name": "typing_extensions",
"specs": []
},
{
"name": "ml_dtypes",
"specs": []
}
],
"lcname": "orbax-checkpoint"
}
JSON
