.. currentmodule:: jax.extend

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

.. automodule:: jax.extend

Modules
-------

.. toctree::
  :maxdepth: 1

  jax.extend.ffi
  jax.extend.linear_util
  jax.extend.mlir
  jax.extend.random