jax.experimental.serialize_executable.deserialize_and_load#

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

从序列化的可执行文件构造一个 jax.stages.Compiled 对象。

参数

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