.. currentmodule:: jax.extend

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

.. automodule:: jax.extend

Modules
-------

.. toctree::
  :maxdepth: 1

  jax.extend.backend
  jax.extend.core
  jax.extend.linear_util
  jax.extend.lowering
  jax.extend.mlir
  jax.extend.pallas
  jax.extend.random
  jax.extend.xla
