Merge pull request #24446 from gnecula:export_doc

PiperOrigin-RevId: 688886756
jax-ml · Oct 23, 2024 · 7ad73e4 · 7ad73e4
2 parents 81bee42 + 7ed65c8
commit 7ad73e4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/docs/jax.export.rst b/docs/jax.export.rst
@@ -28,6 +28,8 @@ Functions
   minimum_supported_calling_convention_version
   maximum_supported_calling_convention_version
   default_export_platform
+  register_pytree_node_serialization
+  register_namedtuple_serialization
 
 Functions related to shape polymorphism
 ---------------------------------------