jax.experimental.serialize_executable
module#
Pickling support for precompiled binaries.
API#
|
Serializes a compiled binary. |
|
Constructs a jax.stages.Compiled from a serialized executable. |
jax.experimental.serialize_executable
module#Pickling support for precompiled binaries.
|
Serializes a compiled binary. |
|
Constructs a jax.stages.Compiled from a serialized executable. |