# 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/52/24/44915f33cbea4cfd35f654a5ba01d248447bc007f1d049dd20bf58592820/orbax_checkpoint-0.11.24.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.24",
"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": "99e8acef62ec5c4b8658eacfce23d3f4e866d3cf5d07c509ca03eebb0c420a6c",
"md5": "a0428aed60320db9805bb42fbe33df51",
"sha256": "a94178c9ba9fd3d6fd8fc511b6a0f34f7d89798bfdb79661e258cd32ada7650b"
},
"downloads": -1,
"filename": "orbax_checkpoint-0.11.24-py3-none-any.whl",
"has_sig": false,
"md5_digest": "a0428aed60320db9805bb42fbe33df51",
"packagetype": "bdist_wheel",
"python_version": "py3",
"requires_python": ">=3.10",
"size": 529268,
"upload_time": "2025-08-28T20:51:47",
"upload_time_iso_8601": "2025-08-28T20:51:47.917358Z",
"url": "https://files.pythonhosted.org/packages/99/e8/acef62ec5c4b8658eacfce23d3f4e866d3cf5d07c509ca03eebb0c420a6c/orbax_checkpoint-0.11.24-py3-none-any.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": null,
"digests": {
"blake2b_256": "522444915f33cbea4cfd35f654a5ba01d248447bc007f1d049dd20bf58592820",
"md5": "ec6155c679521a132ff794c14c689418",
"sha256": "4e7afe927d1ed6d8160bacf5ed4fef56c1370320e0ebdfda213c6351a2e3c0d0"
},
"downloads": -1,
"filename": "orbax_checkpoint-0.11.24.tar.gz",
"has_sig": false,
"md5_digest": "ec6155c679521a132ff794c14c689418",
"packagetype": "sdist",
"python_version": "source",
"requires_python": ">=3.10",
"size": 372123,
"upload_time": "2025-08-28T20:51:49",
"upload_time_iso_8601": "2025-08-28T20:51:49.089303Z",
"url": "https://files.pythonhosted.org/packages/52/24/44915f33cbea4cfd35f654a5ba01d248447bc007f1d049dd20bf58592820/orbax_checkpoint-0.11.24.tar.gz",
"yanked": false,
"yanked_reason": null
}
],
"upload_time": "2025-08-28 20:51:49",
"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"
}
JSON
