safejax

View on PyPIReverse Dependencies (0)

0.5.0 safejax-0.5.0-py3-none-any.whl

Wheel Details

Project: safejax
Version: 0.5.0
Filename: safejax-0.5.0-py3-none-any.whl
Download: [link]
Size: 12675
MD5: 42fe8b80947d03c2baa50a4a9fbce912
SHA256: 182146a12c621f2695e6b927d6b71d4684003c39d757c77a9d5ca8a3b8105a78
Uploaded: 2023-01-05 11:56:54 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: safejax
Version: 0.5.0
Summary: Serialize JAX, Flax, Haiku, or Objax model params with `safetensors`
Author-Email: Alvaro Bartolome <alvarobartt[at]yahoo.com>
Project-Url: Documentation, https://alvarobartt.github.io/safejax
Project-Url: Issues, https://github.com/alvarobartt/safejax/issues
Project-Url: Source, https://github.com/alvarobartt/safejax
Classifier: Development Status :: 4 - Beta
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Science/Research
Classifier: Programming Language :: Python
Classifier: Programming Language :: Python :: 3.7
Classifier: Programming Language :: Python :: 3.8
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Topic :: Software Development :: Libraries
Classifier: Topic :: Software Development :: Libraries :: Python Modules
Requires-Python: >=3.7
Requires-Dist: dm-haiku (~=0.0.9)
Requires-Dist: flax (~=0.6.2)
Requires-Dist: fsspec (~=2022.11.0)
Requires-Dist: jaxlib (~=0.3.25)
Requires-Dist: jax (~=0.3.25)
Requires-Dist: objax (~=1.6.0)
Requires-Dist: safetensors (~=0.2.5)
Requires-Dist: mkdocs-git-revision-date-localized-plugin (~=1.1.0); extra == "docs"
Requires-Dist: mkdocs-material (~=8.5.4); extra == "docs"
Requires-Dist: mkdocstrings[python] (~=0.19.0); extra == "docs"
Requires-Dist: mkdocs (~=1.4.0); extra == "docs"
Requires-Dist: black (~=22.10.0); extra == "quality"
Requires-Dist: pip-tools (~=6.12.0); extra == "quality"
Requires-Dist: pre-commit (~=2.20.0); extra == "quality"
Requires-Dist: ruff (~=0.0.194); extra == "quality"
Requires-Dist: flaxmodels (~=0.1.2); extra == "tests"
Requires-Dist: pytest-lazy-fixture (~=0.6.3); extra == "tests"
Requires-Dist: pytest (~=7.1.2); extra == "tests"
Provides-Extra: docs
Provides-Extra: quality
Provides-Extra: tests
Description-Content-Type: text/markdown
License-Expression: MIT
License-File: LICENSE
[Description omitted; length: 8379 characters]

WHEEL

Wheel-Version: 1.0
Generator: hatchling 1.12.2
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
safejax/__init__.py sha256=S1nCXi8JA_rIdKpME6euxOusvi-Di012af3NPFEUQ30 277
safejax/flax.py sha256=u73IyqWvXqqrGSVSpwXigeLZZ6nuS9-UfnCbAipEl7A 397
safejax/haiku.py sha256=j86vpsFZoJ0GRg5HKaJi1GWXxTcO_qLEA1pb-JxNaEY 316
safejax/objax.py sha256=T3qzOUJ9jJE5GpvaoseObiIZJbFC2h1W8vNL4DOC_kw 3853
safejax/py.typed sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
safejax/typing.py sha256=cso6vBXWQrJrDZEdSCmUstRKGMGLCaoYIwzWX0Kvut4 564
safejax/utils.py sha256=YHf60u1U5uCBE-4f1e9LVKttcIbf6W-HMXnhewe_KEg 6294
safejax/core/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
safejax/core/load.py sha256=HeHuBL4_wyTP5GvUdXts9f8w_hDpmjO-meRAYDuJT0Y 3634
safejax/core/save.py sha256=dIJ-ahIziryqky2sv5J83ktIRXjxexUNZHn2rAPF68Y 2821
safejax-0.5.0.dist-info/METADATA sha256=W2vHCSio2p6bnr1riQl-k54usUV06X9k3oseHpJfnpg 10348
safejax-0.5.0.dist-info/WHEEL sha256=hKi7AIIx6qfnsRbr087vpeJnrVUuDokDHZacPPMW7-Y 87
safejax-0.5.0.dist-info/licenses/LICENSE sha256=pHCbcfMj4Zai8CwBPbjpRztb93igu9pZdCj_CdYYGr8 1104
safejax-0.5.0.dist-info/RECORD