serialize(compiled)

Serializes a compiled binary.

deserialize_and_load(serialized, in_tree, ...)

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