jax.extend.mlir module
jax.extend.mlir module#
bytes) -> str | |
bytes) -> bytes | |
bool = True, validate_static_shapes: bool = True, enable_shardy: bool = False) -> bytes | |
str, use_mixed_serialization: bool = False) -> bytes serialize_portable_artifact(mlir_module: str, target: str, use_mixed_serialization: bool = False) -> bytes |
