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

.. automodule:: jax.extend.linear_util

.. autosummary::
  :toctree: _autosummary

  Callable
  StoreException
  WrappedFun
  cache
  merge_linear_aux
  transformation
  transformation2
  transformation_with_aux
  transformation_with_aux2
  wrap_init
