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

.. automodule:: jax.extend.linear_util

.. autosummary::
  :toctree: _autosummary

  StoreException
  WrappedFun
  cache
  merge_linear_aux
  transformation
  transformation_with_aux
  wrap_init