jax.extend.linear_util.rst

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