diff --git a/docs/jax.experimental.rst b/docs/jax.experimental.rst index 73cea4a40..a4d1f664a 100644 --- a/docs/jax.experimental.rst +++ b/docs/jax.experimental.rst @@ -26,6 +26,7 @@ Experimental Modules jax.experimental.compilation_cache jax.experimental.key_reuse jax.experimental.mesh_utils + jax.experimental.serialize_executable jax.experimental.shard_map Experimental APIs diff --git a/docs/jax.experimental.serialize_executable.rst b/docs/jax.experimental.serialize_executable.rst new file mode 100644 index 000000000..67fa76cf8 --- /dev/null +++ b/docs/jax.experimental.serialize_executable.rst @@ -0,0 +1,13 @@ +``jax.experimental.serialize_executable`` module +================================================ + +.. automodule:: jax.experimental.serialize_executable + +API +--- + +.. autosummary:: + :toctree: _autosummary + + serialize + deserialize_and_load