From 936ec9838055dcd9dc8c1fb89dc6d039a94cdd89 Mon Sep 17 00:00:00 2001 From: Frederic Bastien Date: Tue, 14 May 2024 13:36:35 -0700 Subject: [PATCH] Add doc for jax.experimental.serialize_executable module --- docs/jax.experimental.rst | 1 + docs/jax.experimental.serialize_executable.rst | 13 +++++++++++++ 2 files changed, 14 insertions(+) create mode 100644 docs/jax.experimental.serialize_executable.rst 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