jax.experimental.serialize_executable.serialize#

jax.experimental.serialize_executable.serialize(compiled)[source]#

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

由于 pytrees 不可序列化，因此它们会被返回，以便用户可以正确处理。

参数:

compiled (jax.stages.Compiled)