# 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.9",
"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/cc/e0/f6a15d26f2bf732be60c0df6e267aa28cce7a1d898daea50cc2bd079d801/orbax_checkpoint-0.5.11.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.5.11",
"project_urls": {
"homepage": "http://github.com/google/orbax",
"repository": "http://github.com/google/orbax"
},
"split_keywords": [
"jax machine learning",
" checkpoint",
" training"
],
"urls": [
{
"comment_text": "",
"digests": {
"blake2b_256": "c27f22962db60e8e2765c310a64c1b5df766850a62b1573870703527e1499ed1",
"md5": "6089272660bb9e90eefe99e0aee8f714",
"sha256": "a0950119f6fece0d2e2d84e04c6ca2136b6c2a59ef996927765f8511d8ee4139"
},
"downloads": -1,
"filename": "orbax_checkpoint-0.5.11-py3-none-any.whl",
"has_sig": false,
"md5_digest": "6089272660bb9e90eefe99e0aee8f714",
"packagetype": "bdist_wheel",
"python_version": "py3",
"requires_python": ">=3.9",
"size": 203155,
"upload_time": "2024-05-10T19:41:31",
"upload_time_iso_8601": "2024-05-10T19:41:31.750852Z",
"url": "https://files.pythonhosted.org/packages/c2/7f/22962db60e8e2765c310a64c1b5df766850a62b1573870703527e1499ed1/orbax_checkpoint-0.5.11-py3-none-any.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": "",
"digests": {
"blake2b_256": "cce0f6a15d26f2bf732be60c0df6e267aa28cce7a1d898daea50cc2bd079d801",
"md5": "ab3bde40fe3272ef0ad998e68e759e2d",
"sha256": "64161132c83888c2ccee4db80078914507e5f35ad47bcf5bfc5fa2b07036a2a7"
},
"downloads": -1,
"filename": "orbax_checkpoint-0.5.11.tar.gz",
"has_sig": false,
"md5_digest": "ab3bde40fe3272ef0ad998e68e759e2d",
"packagetype": "sdist",
"python_version": "source",
"requires_python": ">=3.9",
"size": 149581,
"upload_time": "2024-05-10T19:41:33",
"upload_time_iso_8601": "2024-05-10T19:41:33.931990Z",
"url": "https://files.pythonhosted.org/packages/cc/e0/f6a15d26f2bf732be60c0df6e267aa28cce7a1d898daea50cc2bd079d801/orbax_checkpoint-0.5.11.tar.gz",
"yanked": false,
"yanked_reason": null
}
],
"upload_time": "2024-05-10 19:41:33",
"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
