jax.ref module

.. automodule:: jax.ref

:mod:jax.ref has the API for working with :code:ArrayRef.

API

.. autosummary:: :toctree: _autosummary

AbstractRef Ref freeze get new_ref set swap addupdate