jax.experimental.serialize_executable.serialize

内容

jax.experimental.serialize_executable.serialize#

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

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

由于 pytree 不可序列化，因此将其返回以便用户可以正确处理它们。

参数：: compiled (jax.stages.Compiled)