graph
------------------------

.. automodule:: flax.nnx
.. currentmodule:: flax.nnx


.. autofunction:: split
.. autofunction:: merge
.. autofunction:: update
.. autofunction:: pop
.. autofunction:: state
.. autofunction:: variables
.. autofunction:: graph
.. autofunction:: graphdef
.. autofunction:: iter_graph
.. autofunction:: clone
.. autofunction:: call
.. autofunction:: cached_partial

.. autoclass:: GraphDef
  :members:

.. autoclass:: UpdateContext
  :members:

.. autofunction:: update_context
.. autofunction:: current_update_context