PySpect

Home

Lists

Top stars

Rising stars

← Back to orbax-checkpoint

Dependencies for orbax-checkpoint

Frequently asked questions

© 2025-2026 PySpect

Dependencies of orbax-checkpoint (38).
Dependency	Optional	Group
absl-py	false
aiofiles	false
etils	false
humanize	false
jax	false
msgpack	false
nest-asyncio	false
numpy	false
prometheus-client	false
protobuf	false
psutil	false
pyyaml	false
simplejson	false
tensorstore	false
typing-extensions	false
uvloop	false
aiosqlite	true	tiering-service
chex	true	testing
clu	true	docs
fastapi	true	testing
fire	true	tiering-service
flax	true	docs
google-cloud-logging	true	docs
grain	true	docs
greenlet	true	tiering-service
grpcio-tools	true	tiering-service
httpx	true	testing
mock	true	testing
opencv-python	true	docs
pysqlite3	true	tiering-service
pytest	true	testing
pytest-xdist	true	testing
pytimeparse	true	tiering-service
safetensors	true	docs
sqlalchemy	true	tiering-service
tensorflow	true	testing
tensorflow-datasets	true	docs
torch	true	testing

Dependent packages

Packages depending on orbax-checkpoint (6).
Package	Optional	Group
brax	false
flax	false
orbax-export	false
causalml	true	jax
deepmd-kit	true	jax
penzai	true	notebook