# 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/48/ce/4a3386e9e4bc95ab638c2779e259a231ffc2ae1fc2b67a68f4d6d8a794f2/orbax_checkpoint-0.9.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.9.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": "",
"digests": {
"blake2b_256": "8a4119de59b4a8581ad364970b2429481838cea8d190f50bc8f672ca329b64e0",
"md5": "99af54e1384ba4473b55447accc5229d",
"sha256": "d33e23e63b7ffcf66de3fb5daa3d11bf24a34e66d133eb4a037788f22155857e"
},
"downloads": -1,
"filename": "orbax_checkpoint-0.9.1-py3-none-any.whl",
"has_sig": false,
"md5_digest": "99af54e1384ba4473b55447accc5229d",
"packagetype": "bdist_wheel",
"python_version": "py3",
"requires_python": ">=3.10",
"size": 296514,
"upload_time": "2024-11-11T18:56:53",
"upload_time_iso_8601": "2024-11-11T18:56:53.191302Z",
"url": "https://files.pythonhosted.org/packages/8a/41/19de59b4a8581ad364970b2429481838cea8d190f50bc8f672ca329b64e0/orbax_checkpoint-0.9.1-py3-none-any.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": "",
"digests": {
"blake2b_256": "48ce4a3386e9e4bc95ab638c2779e259a231ffc2ae1fc2b67a68f4d6d8a794f2",
"md5": "ca8f40aa2d84954bd68390887298786d",
"sha256": "edf76d8fc482a9a0296645522f6d13ee09c499af0ef2f9369899cbfee31c7f88"
},
"downloads": -1,
"filename": "orbax_checkpoint-0.9.1.tar.gz",
"has_sig": false,
"md5_digest": "ca8f40aa2d84954bd68390887298786d",
"packagetype": "sdist",
"python_version": "source",
"requires_python": ">=3.10",
"size": 212247,
"upload_time": "2024-11-11T18:56:55",
"upload_time_iso_8601": "2024-11-11T18:56:55.127033Z",
"url": "https://files.pythonhosted.org/packages/48/ce/4a3386e9e4bc95ab638c2779e259a231ffc2ae1fc2b67a68f4d6d8a794f2/orbax_checkpoint-0.9.1.tar.gz",
"yanked": false,
"yanked_reason": null
}
],
"upload_time": "2024-11-11 18:56:55",
"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
