orbax-checkpoint has 5.9M monthly downloads and 524 GitHub stars. Version 0.12.1 supports Python 3.11+, first published in 2023.
It is actively maintained under a permissive license, last released in Jun 2026.
It is built on top of absl-py, aiofiles and etils. It is used by 72 tracked packages, including flax, orbax-export and pathwaysutils.
GitHub topics include flax, jax and checkpoint.
Package Insights
((week_daily_avg - month_daily_avg) / month_daily_avg) * 100Weekly Downloads
GitHub Stars
Downloads by OS
Python Versions
Top Countries
Dependencies
- absl-py
- aiofiles
- etils
- humanize
- jax >=0.6.0
- msgpack
- nest-asyncio
- numpy
- prometheus-client >=0.20.0
- protobuf
- psutil
- pyyaml
- simplejson >=3.16.0
- tensorstore >=0.1.84
- typing-extensions
- uvloop
31 optional dependencies
- aiofiles[docs]
- aiofiles[testing]
- aiosqlite[tiering-service]
- chex[testing]
- clu[docs]
- clu[testing]
- fastapi[testing]
- fire[tiering-service]
- flax[docs]
- flax[testing]
- google-cloud-logging[testing]
- google-cloud-logging[docs]
- grain[testing]
- grain[docs]
- greenlet[tiering-service]
- grpcio-tools[tiering-service]
- httpx[testing]
- httpx[tiering-service]
- mock[testing]
- opencv-python[docs]
- pysqlite3[tiering-service]
- pytest[testing]
- pytest-xdist[testing]
- pytimeparse[tiering-service]
- safetensors[docs]
- safetensors[testing]
- sqlalchemy[tiering-service]
- tensorflow[testing]
- tensorflow-datasets[docs]
- torch[testing]
- uvloop[tiering-service]
Used By
- flax
- orbax-export
- pathwaysutils
- google-tunix
- brax
- kauldron
- gemma
- jax2onnx
- jax-ai-stack
- roxxel
- playground
- paxml
- alberta-framework
- lalamo
- mlip
- cd-dynamax
- mintstate
- astronomix
- clip-jax
- datarax
- xax
- avitai-artifex
- xtrax
- synthpix
- apax
- bacpipe
- nadl
- hackable-diffusion
- dynestyx
- ife-surrogate
- alphagenome-ft
- kinetix-env
- tn4ml
- nnx-ppo
- jax-neural-operators
- tensorial
- jaxion
- beastsim
- gensbi
- kithara
- scribe-data
- opifex
- flax-pilot
- finhjb
- jaxnn
- adirondax
- feedback-grape
- julax
- multistage
- nasimjax