跳转到主要内容

Orbax Export

项目描述

Orbax Export

pip install orbax-export (最新PyPi版本) OR

pip install 'git+https://github.com/google/orbax/#subdirectory=export' (从此仓库,在HEAD)

import orbax.export

Orbax为JAX用户提供了一个序列化库,允许将JAX模型导出为TensorFlow SavedModel格式。

请注意,orbax-export需要TensorFlow,但默认情况下不包括它,以允许在版本选择上的灵活性。如果您希望使用标准TensorFlow进行安装,请使用pip install orbax-export[all]

要开始,请查看我们的文档

项目详情


下载文件

下载适合您平台的应用程序。如果您不确定选择哪个,请了解更多关于安装包的信息。

源分布

orbax_export-0.0.5.tar.gz (39.3 kB 查看哈希值)

上传时间:

构建版本

orbax_export-0.0.5-py3-none-any.whl (62.5 kB 查看哈希值)

上传时间 Python 3

由以下支持