[docs] Added two new APIs to the export API docs

jax-ml · Oct 22, 2024 · 7ed65c8 · 7ed65c8
1 parent a2bc8c2
commit 7ed65c8
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
 ---------------------------------------