{"id":13521034,"url":"https://github.com/google/orbax","last_synced_at":"2026-02-27T10:35:31.844Z","repository":{"id":37083898,"uuid":"489442143","full_name":"google/orbax","owner":"google","description":"Orbax provides common checkpointing and persistence utilities for JAX users","archived":false,"fork":false,"pushed_at":"2026-02-24T17:31:42.000Z","size":7542,"stargazers_count":482,"open_issues_count":348,"forks_count":82,"subscribers_count":8,"default_branch":"main","last_synced_at":"2026-02-24T18:16:19.363Z","etag":null,"topics":["checkpoint","flax","jax"],"latest_commit_sha":null,"homepage":"https://orbax.readthedocs.io/","language":"Python","has_issues":true,"has_wiki":null,"has_pages":null,"mirror_url":null,"source_name":null,"license":"apache-2.0","status":null,"scm":"git","pull_requests_enabled":true,"icon_url":"https://github.com/google.png","metadata":{"files":{"readme":"README.md","changelog":"CHANGELOG.md","contributing":"CONTRIBUTING.md","funding":null,"license":"LICENSE","code_of_conduct":null,"threat_model":null,"audit":null,"citation":null,"codeowners":null,"security":null,"support":null,"governance":null,"roadmap":null,"authors":null,"dei":null,"publiccode":null,"codemeta":null,"zenodo":null,"notice":null,"maintainers":null,"copyright":null,"agents":null,"dco":null,"cla":null}},"created_at":"2022-05-06T17:36:22.000Z","updated_at":"2026-02-24T14:32:38.000Z","dependencies_parsed_at":"2023-12-18T20:35:54.251Z","dependency_job_id":"59b42170-504b-40ff-89b0-eadc9b3be13d","html_url":"https://github.com/google/orbax","commit_stats":{"total_commits":554,"total_committers":42,"mean_commits":13.19047619047619,"dds":0.4169675090252708,"last_synced_commit":"43f36618da8b47392525bff325ce7dd8088c34ea"},"previous_names":[],"tags_count":71,"template":false,"template_full_name":null,"purl":"pkg:github/google/orbax","repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/google%2Forbax","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/google%2Forbax/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/google%2Forbax/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/google%2Forbax/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/google","download_url":"https://codeload.github.com/google/orbax/tar.gz/refs/heads/main","sbom_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/google%2Forbax/sbom","scorecard":null,"host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":286080680,"owners_count":29891703,"icon_url":"https://github.com/github.png","version":null,"created_at":"2022-05-30T11:31:42.601Z","updated_at":"2026-02-27T09:48:51.284Z","status":"ssl_error","status_checked_at":"2026-02-27T09:48:43.992Z","response_time":57,"last_error":"SSL_read: unexpected eof while reading","robots_txt_status":"success","robots_txt_updated_at":"2025-07-24T06:49:26.215Z","robots_txt_url":"https://github.com/robots.txt","online":false,"can_crawl_api":true,"host_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub","repositories_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories","repository_names_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repository_names","owners_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners"}},"keywords":["checkpoint","flax","jax"],"created_at":"2024-08-01T06:00:26.863Z","updated_at":"2026-02-27T10:35:31.791Z","avatar_url":"https://github.com/google.png","language":"Python","funding_links":[],"categories":["See also: other libraries in the JAX ecosystem","Python","Libraries"],"sub_categories":[],"readme":"# Orbax\n\n[Orbax](https://orbax.readthedocs.io/en/latest/) provides common checkpointing and persistence utilities for JAX users.\n\n## [Documentation](https://orbax.readthedocs.io/en/latest/)\n\nRefer to our full documentation [here](https://orbax.readthedocs.io/en/latest/).\n\n## Checkpointing\n\n`pip install orbax-checkpoint` (latest PyPi release) OR\n\n`pip install 'git+https://github.com/google/orbax/#subdirectory=checkpoint'`\n(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\u003e [!NOTE]\n\u003e Please see\n[Announcements](https://orbax.readthedocs.io/en/latest/guides/checkpoint/orbax_checkpoint_announcements.html)\nfor important updates.\n\n\n## Exporting\n\n`pip install orbax-export` (latest PyPi release) OR\n\n`pip install 'git+https://github.com/google/orbax/#subdirectory=export'`\n(from this repository, at HEAD)\n\n`import orbax.export`\n\nOrbax also includes a serialization library for JAX users, enabling the\nexporting of JAX models to the TensorFlow SavedModel format.\n\nNote that `orbax-export` requires TensorFlow, but does not include it by default\nto allow for flexibility in version choice. If you wish to install with standard\nTensorFlow, please use `pip install orbax-export[all]`.\n\n\n## Releases\n\nYou can find the latest releases of Orbax Checkpointing at [orbax-checkpoint](https://pypi.org/project/orbax-checkpoint/#history) and for Orbax Export at [orbax-export](https://pypi.org/project/orbax-export/#history).\n\n\n## Support\n\nContact orbax-dev@google.com for help or with any questions about Orbax!\n\n### History\n\nOrbax was initially published as a catch-all package itself. In order to\nminimize dependency bloat for users, we have frozen that package at\n`orbax-0.1.6`, and will continue to release future changes under the\ndomain-specific utilities detailed above (e.g. `orbax-checkpoint`).\n\nAs we have preserved the orbax namespace, existing import statements can remain\nunchanged (e.g. `from orbax import checkpoint`).\n","project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fgoogle%2Forbax","html_url":"https://awesome.ecosyste.ms/projects/github.com%2Fgoogle%2Forbax","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fgoogle%2Forbax/lists"}