jax.extend.lowering.upstream_dialects# jax.extend.lowering.upstream_dialects = <jaxlib.mlir._mlir_libs._mlir.ir.DialectRegistry object>#