jax.experimental.serialize_executable.serialize#

jax.experimental.serialize_executable.serialize(compiled)[源代码]#

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

因为 pytrees 是不可序列化的，所以它们会被返回，以便用户可以正确处理它们。

参数：

compiled (jax.stages.Compiled)