jax.experimental.serialize_executable.deserialize_and_load

jax.experimental.serialize_executable.deserialize_and_load#

jax.experimental.serialize_executable.deserialize_and_load(serialized, in_tree, out_tree, backend=None)[source]#

从序列化可执行文件中构建 jax.stages.Compiled。

参数:

backend (str | xc.Client | None | None)