jax.export 模块

`jax.export` 模块#

jax.export 是一个用于导出和序列化 JAX 函数以进行持久存档的库。

请参阅导出文档。

`Exported`(fun_name, in_tree, in_avals, ...)	一个降低到 StableHLO 的 JAX 函数。
`DisabledSafetyCheck`(_impl)	在（反）序列化时应跳过的安全检查。

`export`(fun_jit, *[, platforms, ...])	导出一个用于持久序列化的 JAX 函数。
`deserialize`(blob)	反序列化一个导出的对象。
`minimum_supported_calling_convention_version`	int([x]) -> 整数 int(x, base=10) -> 整数
`maximum_supported_calling_convention_version`	int([x]) -> 整数 int(x, base=10) -> 整数
`default_export_platform`()	检索默认的导出平台。

`symbolic_shape`(shape_spec, *[, constraints, ...])	从字符串表示构建一个符号形状。
`symbolic_args_specs`(args, shapes_specs[, ...])	为 export 构建一个 jax.ShapeDtypeSpec 参数规格的 pytree。
`is_symbolic_dim`(p)	检查一个维度是否为符号。
`SymbolicScope`([constraints_str])	标识符号表达式的范围。