jax.export 模块

内容

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

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

`export`(fun_jit, *[, platforms, ...])	导出 JAX 函数以进行持久序列化。
`deserialize`(blob)	反序列化 Exported 对象。
`minimum_supported_calling_convention_version`	int([x]) -> integer int(x, base=10) -> integer
`maximum_supported_calling_convention_version`	int([x]) -> integer int(x, base=10) -> integer
`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])	识别符号表达式的范围。