orbax-export

View on PyPIReverse Dependencies (1)

0.0.6 orbax_export-0.0.6-py3-none-any.whl

Wheel Details

Project: orbax-export
Version: 0.0.6
Filename: orbax_export-0.0.6-py3-none-any.whl
Download: [link]
Size: 86213
MD5: 89a6af22dc2def7fa18525c53b4608ad
SHA256: ec7041c7ef92d25ad71a1867a857de461eb574b245d5e8f550b4dcf23438176a
Uploaded: 2024-11-15 02:26:40 +0000

dist-info

METADATA

Metadata-Version: 2.3
Name: orbax-export
Version: 0.0.6
Summary: Orbax Export
Author-Email: Orbax Authors <orbax-dev[at]google.com>
Project-Url: homepage, http://github.com/google/orbax
Project-Url: repository, http://github.com/google/orbax
Keywords: JAX machine learning,serialization,export
Classifier: Development Status :: 4 - Beta
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Programming Language :: Python
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Python: >=3.10
Requires-Dist: absl-py
Requires-Dist: dataclasses-json
Requires-Dist: etils
Requires-Dist: jax (>=0.4.34)
Requires-Dist: jaxlib
Requires-Dist: jaxtyping
Requires-Dist: numpy
Requires-Dist: orbax-checkpoint (>=0.9.0)
Requires-Dist: tensorflow (>=2.15.0); extra == "all"
Requires-Dist: pytest; extra == "testing"
Requires-Dist: pytest-xdist; extra == "testing"
Requires-Dist: tf-nightly; extra == "testing"
Requires-Dist: requests; extra == "testing"
Requires-Dist: chex; extra == "testing"
Provides-Extra: all
Provides-Extra: testing
Description-Content-Type: text/markdown
[Description omitted; length: 656 characters]

WHEEL

Wheel-Version: 1.0
Generator: flit 3.10.1
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
orbax/export/__init__.py sha256=tFWzlq25nlHM2wHAVX8CXHBgFFK6VwlEqBUdi6LjYZk 1688
orbax/export/config.py sha256=Qc4YjP_Chqdtgj2ABBQ4HaE1EFDnr0FGYufFAXEGWC8 7050
orbax/export/config_test.py sha256=M20DkcttK5vj__HyyuHIeF7KX54d1D2tNys8ciTRlxM 2710
orbax/export/conftest.py sha256=ghkxmpxGtLnqqD7-UeNmnpTo9didMaWa1Q9sDHnhf4Y 740
orbax/export/constants.py sha256=BZ5bboyshd5Jou2BcqQojdRkr5hctW-6jsC-BhCyXpU 2847
orbax/export/dtensor_export_test.py sha256=7csjpieCQu-SQf9yUnR3PLBkSBUx0Gy2OPtE4VHsdAk 3959
orbax/export/dtensor_initialization_test.py sha256=gsRcb8TH7H3R258Y6vLLZHsV_PgGQLRIhw8X1CJ273s 1389
orbax/export/dtensor_utils.py sha256=OvEyehZ4_8Gnij9wD_tdX2Kvmwo0xJH65SStNgXnxWI 11487
orbax/export/dtensor_utils_test.py sha256=86mnaXF0E-mc4FgckAD-Ld4h4JBu0aRx24V2jyEj08M 8291
orbax/export/export_base.py sha256=Fqg40WCGpsZM0QPiM98TFx7cqg00epHGgnItRgzoBdU 1804
orbax/export/export_manager.py sha256=H9HEQFI41HdQWlOCOOla-7WDxn60Gzw7i4Vy-sdZZvo 7905
orbax/export/export_manager_obm_test.py sha256=ClUzLS01Xv5QKoUmTaGu4a1DaGRszAHBWRMhOJL9Auc 2277
orbax/export/export_manager_test.py sha256=fnY6R5sSSLH_e0bJCluLybXPybZFFvIzuT2mzmBj6O8 17270
orbax/export/jax_module.py sha256=DlZVcQ_CzQJ478MRk48WRvq1Vf9mpAnZ2V2m26i6WsE 10015
orbax/export/jax_module_test.py sha256=4ECtqO2QPok-Cq3MvVFCKToEKzwOrqSpdLgjQJc9bPo 5471
orbax/export/obm_export.py sha256=MrfI6pwaJ722x9FvN2cZKMHqtvgEjavXx-FjahK26_0 2279
orbax/export/obm_export_test.py sha256=Kc6nRUGCSPwpJwNhsrOkD4hJcGLKCaRnWgCxfbfYMDs 1517
orbax/export/serving_config.py sha256=6CDMAd1cO_gCx9gWlDOxVFoOb0KRg3nEHnYiZ4DWF_g 7661
orbax/export/serving_config_test.py sha256=CBl3bpBitOiQphS7Dy2n8QHOpF9vH0kS-beQ7It8N0Q 1434
orbax/export/tensorflow_export.py sha256=G-BwRZRRylxsiBjogQXCc1iwA_FgfeZnScBSS3Qtr2s 2420
orbax/export/tensorflow_export_test.py sha256=2DduXUxaRw22p0GUF_wgvBHWxy-KSTZtz6ttKPqGFMA 2104
orbax/export/typing.py sha256=BfiuA9DYffCMP7j_y4QaVwuf4m_973SqVc_yQsIz49U 1130
orbax/export/utils.py sha256=cY2sQ7vPS5snMkfhLKzKC75yZ_Gm7I6Cj9QgNmeGBdY 16049
orbax/export/utils_test.py sha256=LhDL38R9Z4zNohs4K_o1pztXFiQCX4UMDmoTLgG_1bM 6706
orbax/export/examples/__init__.py sha256=4Pb4JEUzytOA9FMwntBJ8un2Fmf1ARNNgHbC3tKnYqo 583
orbax/export/examples/flax_mnist_main.py sha256=lt393ELx-KcYczDtlsU6MEARnjqUWBh6-oP-2Ad7x4c 3103
orbax/export/examples/flax_mnist_main_test.py sha256=BtYfaCUhWw4KOc9RQy2gubKQNfcxPitG4YJCH9qVhRw 1549
orbax/export/examples/periodically_export_main.py sha256=AZoaHLODNuA_Ve2cCD-zkUiuy1hpMtiwsibXJEhWYrY 4770
orbax/export/examples/periodically_export_main_test.py sha256=bQBy7LqDcT7FIjp0jvaMPGVaUH_tRVWb3-XdhpwrHJk 1703
orbax/export/modules/obm_module.py sha256=B4k24iCAU19jFEuSrEWdAmCakXRb7wGMprU6gvS_pnc 7223
orbax/export/modules/obm_module_test.py sha256=nWmVDH6QkLUYS5wA4dhTsFL0B07IWf_rlsMNmmhbWds 1721
orbax/export/modules/orbax_module_base.py sha256=gKn8cW2s7ltxgjxaouVa-pqMsJBIsTpKhcycEfhuRtQ 2200
orbax/export/modules/tensorflow_module.py sha256=ksXg5GuQg8KIENZi3Sy2d4kQizitqqUdawtwGg-MdYs 18034
orbax/export/modules/tensorflow_module_test.py sha256=N16_NnkEbixXQyzvNdfQsqOXxoMAOQMtP8LnjTLIsNM 16966
orbax/export/validate/__init__.py sha256=Pf8dpKUQKu9-yyTGs_75UM3OgBDXUMfK-X-IhxBanpY 810
orbax/export/validate/validation_job.py sha256=xxy_PKkDDSyc1NNRCTMQy5hYnLYuOfAqwcHQFdMpAyo 3872
orbax/export/validate/validation_job_test.py sha256=M6rAz3SkGPdiYcJUABjw3ExJSg7Bl6fqfhWP1u5h8Vw 3313
orbax/export/validate/validation_manager.py sha256=aatXWv8t8iQWYx-uEuC8O621yiP20s4GtG3J-LOhD9I 8052
orbax/export/validate/validation_manager_test.py sha256=DUDw0tRr6D_T82W9hGX9_mCKm8_1EXt7wOES5Dtod10 10231
orbax/export/validate/validation_report.py sha256=8IqrY2oWBWxosQhJWYQ_Iu8eNtsAGA4hleLaqnP59OE 8865
orbax/export/validate/validation_report_test.py sha256=WaB4FPSKXeTzu9Pq9FJXRmzR_Vfy7WjQmN6kCB-D2xM 5602
orbax/export/validate/validation_utils.py sha256=q5cB9dbyUaTCgnh4MaQdCg44nwx6LJvSmA03O0U6vkk 2301
orbax/export/validate/validation_utils_test.py sha256=4H93aa5LCKH8TcNYcKs6KBLPqrsNO24AoL-cH5sXoLg 2628
orbax_export-0.0.6.dist-info/LICENSE sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0 11357
orbax_export-0.0.6.dist-info/WHEEL sha256=CpUCUxeHQbRN5UGRQHYRJorO5Af-Qy_fHMctcQ8DSGI 82
orbax_export-0.0.6.dist-info/METADATA sha256=24Bmyfy_y-_5IhJaiJpWs7mqTkr0-tmRZWVq09hilak 1859
orbax_export-0.0.6.dist-info/RECORD