jax.experimental.serialize_executable 模块

提供对预编译二进制文件的序列化（Pickling）支持。

API

serialize(compiled)	序列化一个已编译的二进制文件。
deserialize_and_load(serialized, in_tree, ...)	从序列化的可执行文件构建一个 jax.stages.Compiled 对象。