jax.experimental.serialize_executable 模块

内容

jax.experimental.serialize_executable 模块#

预编译二进制文件的腌制支持。

API#

serialize(compiled)

序列化编译后的二进制文件。

deserialize_and_load(serialized, in_tree, ...)

从序列化可执行文件中构造 jax.stages.Compiled。