Data

Tools

Indexes

Applications

Experiments

Summary

An open API service for producing an overview of a list of open source projects.

https://github.com/google/orbax

checkpoint flax jax

Score: 28.79490110828725

Last synced: about 4 hours ago
JSON representation

Repository metadata:

Orbax provides common checkpointing and persistence utilities for JAX users

Owner metadata:

GitHub Events

Total
Last Year

Committers metadata

Last synced: 2 days ago

Total Commits: 2,878
Total Committers: 116
Avg Commits per committer: 24.81
Development Distribution Score (DDS): 0.731

Commits in past year: 906
Committers in past year: 54
Avg Commits per committer in past year: 16.778
Development Distribution Score (DDS) in past year: 0.779

Name Email Commits
Colin Gaffney c****y@g****m 774
Orbax Authors n****y@g****m 522
Adam Cogdell a****l@g****m 197
Niket Kumar Bhumihar n****b@g****m 167
Daniel Ng d****g@g****m 156
Mridul Sahu m****u@g****m 147
Yaning Liang y****g@g****m 93
Angel Mau a****u@g****m 74
Anastasia Petrushkina d****a@g****m 64
He Jiang h****e@g****m 59
Qidi Chen q****n@g****m 57
Yunjie Xu y****u@g****m 38
Nikhil Bansal n****l@g****m 37
Abhishek Agrawal a****r@g****m 36
John Zhang j****g@g****m 35
Oleg Shaldybin o****h@g****m 29
Justin Pan j****n@g****m 26
Yash Katariya y****a@g****m 25
Shutong Li l****g@g****m 23
Zach Meyers z****s@g****m 22
Dinghua Li d****a@g****m 20
Ayush Jain a****h@g****m 18
Peng Wang w****g@g****m 18
Abhinav Singh a****g@g****m 17
Sujeeth Jinesh s****h@g****m 17
Marco Berlot m****t@g****m 16
Jake VanderPlas v****s@g****m 16
Peter Hawkins p****s@g****m 14
Hana Joo h****o@g****m 9
Katherine Wu k****u@g****m 7
and 86 more...

Issue and Pull Request metadata

Last synced: 2 days ago

Total issues: 176
Total pull requests: 4,769
Average time to close issues: 4 months
Average time to close pull requests: 6 days
Total issue authors: 111
Total pull request authors: 44
Average comments per issue: 3.02
Average comments per pull request: 0.2
Merged pull request: 2,663
Bot issues: 19
Bot pull requests: 4,648

Past year issues: 30
Past year pull requests: 1,146
Past year average time to close issues: about 2 months
Past year average time to close pull requests: 7 days
Past year issue authors: 20
Past year pull request authors: 20
Past year average comments per issue: 2.33
Past year average comments per pull request: 0.26
Past year merged pull request: 721
Past year bot issues: 5
Past year bot pull requests: 1,096

More stats: https://issues.ecosyste.ms/repositories/lookup?url=https://github.com/google/orbax

Top Issue Authors

  • copybara-service[bot] (14)
  • hr0nix (7)
  • sshleifer (6)
  • PhilipVinc (5)
  • qGentry (5)
  • github-actions[bot] (5)
  • evelyn22chen (4)
  • ethanluoyc (4)
  • hrbigelow (4)
  • gianlucadetommaso (3)
  • minotru (3)
  • kvablack (3)
  • anveshjayakrishna (3)
  • haohuanw (2)
  • ZaberKo (2)

Top Pull Request Authors

  • copybara-service[bot] (4,643)
  • mridul-sahu (27)
  • findmyway (10)
  • cpgaffney1 (9)
  • gspschmid (8)
  • jakevdp (6)
  • keshavb96 (5)
  • dependabot[bot] (5)
  • hartikainen (4)
  • deepikarajani24 (4)
  • quoctruong (3)
  • ethanluoyc (3)
  • JesseFarebro (3)
  • samos123 (2)
  • lshariprasad (2)

Top Issue Labels

  • checkpoint (107)
  • type:bug (64)
  • type:support (30)
  • type:feature (21)
  • dependencies (10)
  • type:build/install (9)
  • type:performance (6)
  • continuous-integration (5)
  • documentation (5)
  • type:others (5)
  • bug (5)
  • wontfix (4)
  • export (4)
  • duplicate (2)
  • better_errors (1)

Top Pull Request Labels

  • pull ready (32)
  • dependencies (4)
  • ci (1)
  • priority-low (1)
  • enhancement (1)
  • maintenance (1)

Package metadata

pypi.org: orbax-checkpoint

Orbax Checkpoint

  • Homepage:
  • Documentation: https://orbax-checkpoint.readthedocs.io/
  • Licenses: Apache Software License
  • Latest release: 0.12.6 (published 5 days ago)
  • Last Synced: 2026-09-30T04:29:49.063Z (1 day ago)
  • Versions: 113
  • Dependent Packages: 16
  • Dependent Repositories: 313
  • Downloads: 2,456,379 Last month
  • Docker Downloads: 25,097,031
  • Rankings:
    • Downloads: 0.339%
    • Dependent repos count: 0.842%
    • Dependent packages count: 1.04%
    • Average: 1.346%
    • Docker downloads count: 3.164%
  • Maintainers (1)
nixpkgs-unstable: python314Packages.orbax-checkpoint

Orbax provides common utility libraries for JAX users

proxy.golang.org: github.com/google/orbax

  • Homepage:
  • Documentation: https://pkg.go.dev/github.com/google/orbax#section-documentation
  • Licenses: apache-2.0
  • Latest release: v0.12.6 (published 5 days ago)
  • Last Synced: 2026-09-30T04:03:31.205Z (1 day ago)
  • Versions: 82
  • Dependent Packages: 0
  • Dependent Repositories: 0
  • Rankings:
    • Dependent packages count: 6.472%
    • Average: 6.69%
    • Dependent repos count: 6.908%
pypi.org: orbax-export

Orbax Export

  • Homepage:
  • Documentation: https://orbax-export.readthedocs.io/
  • Licenses: Apache Software License
  • Latest release: 0.0.8 (published about 1 year ago)
  • Last Synced: 2026-09-30T04:03:35.043Z (1 day ago)
  • Versions: 9
  • Dependent Packages: 1
  • Dependent Repositories: 1
  • Downloads: 303,229 Last month
  • Rankings:
    • Dependent packages count: 4.776%
    • Downloads: 8.716%
    • Average: 11.695%
    • Dependent repos count: 21.593%
  • Maintainers (1)
spack.io: py-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.

  • Homepage: https://github.com/google/orbax
  • Licenses: []
  • Latest release: 0.5.3 (published over 2 years ago)
  • Last Synced: 2026-09-30T04:03:07.534Z (1 day ago)
  • Versions: 1
  • Dependent Packages: 1
  • Dependent Repositories: 0
  • Rankings:
    • Dependent repos count: 0.0%
    • Average: 27.807%
    • Dependent packages count: 55.613%
anaconda.org: orbax-export

Orbax includes a serialization library for JAX users, enabling the exporting of JAX models to the TensorFlow SavedModel format.

  • Homepage: https://github.com/google/orbax
  • Licenses: Apache-2.0
  • Latest release: 0.0.8 (published 3 months ago)
  • Last Synced: 2026-06-30T16:11:08.983Z (3 months ago)
  • Versions: 1
  • Dependent Packages: 0
  • Dependent Repositories: 0
  • Rankings:
    • Dependent packages count: 34.629%
    • Average: 36.194%
    • Dependent repos count: 37.758%
anaconda.org: 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.

  • Homepage: https://github.com/google/orbax
  • Licenses: Apache-2.0
  • Latest release: 0.11.18 (published 7 months ago)
  • Last Synced: 2026-03-23T10:12:37.095Z (6 months ago)
  • Versions: 1
  • Dependent Packages: 0
  • Dependent Repositories: 0
  • Downloads: 36 Total
  • Rankings:
    • Dependent packages count: 38.984%
    • Average: 40.745%
    • Dependent repos count: 42.506%
nixpkgs-unstable: python314Packages.orbax-export

Serialization library for JAX users, enabling the exporting of JAX models to the TensorFlow SavedModel format

nixpkgs-24.05: python312Packages.orbax-checkpoint

Orbax provides common utility libraries for JAX users

nixpkgs-unstable: python313Packages.orbax-export

Serialization library for JAX users, enabling the exporting of JAX models to the TensorFlow SavedModel format

nixpkgs-24.11: python312Packages.orbax-checkpoint

Orbax provides common utility libraries for JAX users

nixpkgs-unstable: python313Packages.orbax-checkpoint

Orbax provides common utility libraries for JAX users

nixpkgs-24.11: python311Packages.orbax-checkpoint

Orbax provides common utility libraries for JAX users

nixpkgs-24.05: python311Packages.orbax-checkpoint

Orbax provides common utility libraries for JAX users

Dependencies

.github/workflows/auto-publish.yml actions
  • etils-actions/pypi-auto-publish v1 composite
.github/workflows/build.yml actions
  • actions/checkout v2 composite
  • actions/setup-python v1 composite
  • styfle/cancel-workflow-action 0.8.0 composite
checkpoint/pyproject.toml pypi
  • absl-py *
  • etils [epath,epy]
  • jax >= 0.4.9
  • jaxlib *
  • msgpack *
  • nest_asyncio *
  • numpy *
  • protobuf *
  • pyyaml *
  • tensorstore >= 0.1.35
  • typing_extensions *
docs/requirements/requirements-docs.txt pypi
  • IPython ==7.16.3
  • absl-py *
  • cached_property *
  • docutils ==0.16
  • etils *
  • importlib_resources *
  • ipykernel ==5.3.4
  • jax *
  • jaxlib *
  • msgpack *
  • myst_nb ==0.13.1
  • nest_asyncio *
  • numpy *
  • pyyaml *
  • sphinx ==4.5.0
  • sphinx-autodoc-typehints ==1.11.1
  • sphinx-book-theme ==0.3.3
  • sphinx-design *
  • sphinxcontrib-bibtex ==2.4.2
  • sphinxcontrib-katex ==0.9.0
  • tensorflow *
  • tensorstore ==0.1.35
  • typing_extensions *
export/pyproject.toml pypi
  • absl-py *
  • dataclasses-json *
  • etils *
  • jax *
  • jaxlib *
  • numpy *
  • orbax-checkpoint *
.github/workflows/build_and_push_docker.yml actions
  • actions/checkout v5 composite
  • docker/build-push-action v6 composite
  • docker/setup-buildx-action v3.11.1 composite
.github/workflows/multiprocess_tests.yml actions
  • actions/checkout 11bd71901bbe5b1630ceea73d27597364c9af683 composite
  • actions/github-script v7 composite
  • actions/setup-python 0b93645e9fea7318ecaed2b359559ac225c90a2b composite
.github/workflows/auto-publish-dry-run.yml actions
  • etils-actions/pypi-auto-publish v1 composite
.github/workflows/auto_label.yml actions
  • actions/github-script v8 composite
.github/workflows/build_image.yml actions
  • actions/checkout v5 composite
  • docker/build-push-action v6 composite
  • docker/setup-buildx-action v3.11.1 composite
pyproject.toml pypi
.github/workflows/lint.yml actions
  • actions/checkout v6 composite
  • actions/setup-python v6 composite
checkpoint/orbax/checkpoint/_src/testing/oss/Dockerfile docker
  • $BASE_IMAGE latest build