跳转到主要内容

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]

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

由以下支持