``jax.extend.xla`` module
=========================

.. automodule:: jax.extend.xla

.. autosummary::
  :toctree: _autosummary

  PipelineStage
  register_hlo_module_transformation