jax.experimental.serialize_executable.deserialize_and_load#

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

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

Parameters:

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