PyPI page
Author:
Alvaro Bartolome
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:
0
Downloads last week:
16
Downloads last month:
30
7-day smoothing