jax.experimental.serialize_executable.deserialize_and_load

jax.experimental.serialize_executable.deserialize_and_load(serialized, in_tree, out_tree, backend=None, execution_devices=None)

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

参数:

• backend (str | xc.Client | None)

• execution_devices (Sequence[xc.Device] | None)