jax.experimental.serialize_executable.serialize

jax.experimental.serialize_executable.serialize(compiled)

Serializes a compiled binary.

Because pytrees are not serializable, they are returned so that the user can handle them properly.

Parameters:

compiled (jax.stages.Compiled)