diff --git a/docs/jax.export.rst b/docs/jax.export.rst index d458b6c64e8e..2095758dd3b3 100644 --- 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 ---------------------------------------