jax.experimental.serialize_executable.serialize

jax.experimental.serialize_executable.serialize(compiled)

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

因为 pytrees 是不可序列化的，它们被返回以便用户可以适当地处理它们。

参数:

compiled (jax.stages.Compiled)