{"categories":[{"label":"Artificial Intelligence","url":"https://skillfed.io/packages/category/scientific-engineering-artificial-intelligence/5"}],"enrichment":{"capability":"Orbax Export serializes JAX models to TensorFlow SavedModel format, enabling JAX-trained models to be deployed and used in TensorFlow environments.","skillfed_tags":["jax-tensorflow-interop","model-serialization"],"use_cases":["Export a trained JAX model to SavedModel format for deployment in TensorFlow-based inference systems.","Share JAX models with teams or collaborators who work primarily in TensorFlow environments.","Convert research models from JAX to a standardized format for production deployment pipelines.","Integrate JAX training workflows with existing TensorFlow-based model serving infrastructure."],"what_it_does":"Orbax Export is a serialization library that bridges JAX and TensorFlow by converting JAX models into TensorFlow's SavedModel format. This enables JAX users to export trained models for deployment in TensorFlow-based production systems or to share models across frameworks. The package is part of the larger Orbax ecosystem and integrates with orbax-checkpoint for model persistence.\n\nThe package requires nine runtime dependencies centered on jax, jaxlib, jaxtyping, numpy, and serialization tools like protobuf and dataclasses-json. TensorFlow is not bundled by default to allow flexibility in version selection\u2014users must either install the optional [all] extra or manage their own TensorFlow installation. The library targets Python 3.10 or later.","worth_installing":"Yes, if you train models in JAX and need to deploy them in TensorFlow environments or share them across frameworks. The package is actively maintained, has no known vulnerabilities, and uses a permissive license. The main consideration is ensuring TensorFlow is installed separately and that your models fit the supported export patterns\u2014check the documentation before committing to a production workflow."},"id":"orbax-export","links":{"html":"https://skillfed.io/packages/orbax-export","md":"https://skillfed.io/packages/orbax-export.md","pypi":"https://pypi.org/project/orbax-export/"},"maintenance":{"status":"active"},"meta":{"latest_release":"2025-09-17","license_spdx":null,"license_treatment":"permissive","name":"orbax-export","python_support":"supports_current","summary":"Orbax Export"},"popularity":{"monthly_downloads":503647,"position":6302,"tier":"top_15000"},"security":{"n_vulnerabilities":0},"version":"0.0.8"}