``jax.export`` module
=====================

.. automodule:: jax.export

:mod:`jax.export` is a library for exporting and serializing JAX functions
for persistent archival.

See the :ref:`export` documentation.

Classes
-------

.. autosummary::
  :toctree: _autosummary

.. autoclass:: Exported
  :members:

.. autoclass:: DisabledSafetyCheck
  :members:

Functions
---------

.. autosummary::
  :toctree: _autosummary

  export
  deserialize
  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
---------------------------------------

.. autosummary::
  :toctree: _autosummary

  symbolic_shape
  symbolic_args_specs
  is_symbolic_dim
  SymbolicScope

Constants
---------

.. data:: jax.export.minimum_supported_serialization_version

   The minimum supported serialization version; see :ref:`export-calling-convention-version`.

.. data:: jax.export.maximum_supported_serialization_version

   The maximum supported serialization version; see :ref:`export-calling-convention-version`.