jax.experimental.serialize_executable.serialize#

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

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

由于 pytrees 是不可序列化的，因此会返回它们，以便用户可以妥善处理。

参数:

compiled (jax.stages.Compiled)