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