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, execution_devices=None)[source]#

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

Warning

It is not safe to call this API with untrusted inputs. Do not do this. Calling this API loads a serialized executable. Even loading such an executable may run arbitrary code on your machine. It is not safe to pass untrusted data here and likely never will be.

Parameters:

  • backend (str | xc.Client | None)

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