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

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


.. autofunction:: dataclass
.. autofunction:: field