jax.experimental.serialize_executable.deserialize_and_load

Contents

jax.experimental.serialize_executable.deserialize_and_load#

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

Constructs a jax.stages.Compiled from a serialized executable.

Parameters:

backend (str | xc.Client | None | None)
execution_devices (Sequence[xc.Device] | None | None)