jax.extend.lowering 模块

JaxIrContext(*args, **kwargs)
LoweringRuleContext(module_context, ...[, ...])	用于 MLIR Lowering 的规则级上下文信息。
upstream_dialects (上游方言)