# 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/18/07/367e728c69dbb0dcf51bc9b1fd4476d9e31eb37aab763698991fabc0c2c7/orbax_checkpoint-0.11.19.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.19",
"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": "52697f0e6bc634a22c5085e225fbad3e8df34d42a37458be3f4984d8eb6a4048",
"md5": "d087396ac77ce8cb16bbdfcebd472004",
"sha256": "f9a9b007db8b7e7175d5be30afd754c4a29cae6d4a631cbe34cb9eb5b6cbd96f"
},
"downloads": -1,
"filename": "orbax_checkpoint-0.11.19-py3-none-any.whl",
"has_sig": false,
"md5_digest": "d087396ac77ce8cb16bbdfcebd472004",
"packagetype": "bdist_wheel",
"python_version": "py3",
"requires_python": ">=3.10",
"size": 490278,
"upload_time": "2025-07-09T01:34:30",
"upload_time_iso_8601": "2025-07-09T01:34:30.546599Z",
"url": "https://files.pythonhosted.org/packages/52/69/7f0e6bc634a22c5085e225fbad3e8df34d42a37458be3f4984d8eb6a4048/orbax_checkpoint-0.11.19-py3-none-any.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": null,
"digests": {
"blake2b_256": "1807367e728c69dbb0dcf51bc9b1fd4476d9e31eb37aab763698991fabc0c2c7",
"md5": "6ff29b6672d9d9377cca96c19683824c",
"sha256": "6f7f2cd89644e1892f2adf17dfa2ad29253712b22374ca9d6b02129922d46795"
},
"downloads": -1,
"filename": "orbax_checkpoint-0.11.19.tar.gz",
"has_sig": false,
"md5_digest": "6ff29b6672d9d9377cca96c19683824c",
"packagetype": "sdist",
"python_version": "source",
"requires_python": ">=3.10",
"size": 346028,
"upload_time": "2025-07-09T01:34:32",
"upload_time_iso_8601": "2025-07-09T01:34:32.110608Z",
"url": "https://files.pythonhosted.org/packages/18/07/367e728c69dbb0dcf51bc9b1fd4476d9e31eb37aab763698991fabc0c2c7/orbax_checkpoint-0.11.19.tar.gz",
"yanked": false,
"yanked_reason": null
}
],
"upload_time": "2025-07-09 01:34:32",
"github": true,
"gitlab": false,
"bitbucket": false,
"codeberg": false,
"github_user": "google",
"github_project": "orbax",
"travis_ci": false,
"coveralls": false,
"github_actions": true,
"lcname": "orbax-checkpoint"
}