jax.extend.lowering module

JaxIrContext(*args, **kwargs)
LoweringRuleContext(module_context, ...[, ...])	Per-rule context information for MLIR lowering.
upstream_dialects