jax.extend.mlir module

deserialize_portable_artifact	bytes) -> str
dialects
hlo_to_stablehlo	bytes) -> bytes
ir
passmanager
refine_polymorphic_shapes	bool = True, validate_static_shapes: bool = True, enable_shardy: bool = False) -> bytes
serialize_portable_artifact	str, use_mixed_serialization: bool = False) -> bytes serialize_portable_artifact(mlir_module: str, target: str, use_mixed_serialization: bool = False) -> bytes