rocm_jax/docs/jax.extend.mlir.rst

``jax.extend.mlir`` module
============================

.. automodule:: jax.extend.mlir

.. autosummary::
  :toctree: _autosummary

  dialects
  ir
  passmanager
Add jax.extend.mlir. Some users of JAX want to use the MLIR dialects defined in jaxlib. In particular, these need to be used by custom lowering rules. Add a semi-public (jax.extend) API to access these, rather than having them use jax._src.lib.mlir. PiperOrigin-RevId: 588448489 2023-12-06 09:16:04 -08:00			``jax.extend.mlir`` module
			`============================`

			`.. automodule:: jax.extend.mlir`

			`.. autosummary::`
			`:toctree: _autosummary`

			`dialects`
			`ir`
			`passmanager`