mirror of
https://github.com/ROCm/jax.git
synced 2025-04-18 21:06:06 +00:00
47f1b3de2c
The rendered documentation is at https://jax--21976.org.readthedocs.build/en/21976/export/export.html#developer-documentation (for the export developer documentation, including compatibility) and https://jax--21976.org.readthedocs.build/en/21976/export/shape_poly.html#debugging (for the shape polymorphism debugging documentation) While testing the compatibility mechanism I discovered that it can be circumvented by caches. To fix this, I added export_ignore_forward_compatibility to mlir.LoweringParameters.
53 lines
1.0 KiB
ReStructuredText
53 lines
1.0 KiB
ReStructuredText
``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
|
|
|
|
Exported
|
|
DisabledSafetyCheck
|
|
|
|
Functions
|
|
---------
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
export
|
|
deserialize
|
|
minimum_supported_calling_convention_version
|
|
maximum_supported_calling_convention_version
|
|
default_export_platform
|
|
|
|
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`.
|