jax.extend.mlir module

.. automodule:: jax.extend.mlir

.. autosummary:: :toctree: _autosummary

deserialize_portable_artifact dialects hlo_to_stablehlo ir passmanager refine_polymorphic_shapes serialize_portable_artifact