safejax

PyPI page
Home page
Author: None
Summary: Serialize JAX, Flax, Haiku, or Objax model params with `safetensors`
Latest version: 0.5.0
Required dependencies: dm-haiku | flax | fsspec | jax | jaxlib | objax | safetensors
Optional dependencies: black | flaxmodels | mkdocs | mkdocs-git-revision-date-localized-plugin | mkdocs-material | mkdocstrings | pip-tools | pre-commit | pytest | pytest-lazy-fixture | ruff

Downloads last day: 3
Downloads last week: 7
Downloads last month: 69