jax.experimental.serialize_executable.serialize#

jax.experimental.serialize_executable.serialize(compiled)[source]#

Serializes a compiled binary.

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

Parameters:

compiled (jax.stages.Compiled)