module
------------------------

.. automodule:: flax.nnx
    :members: iter_children, iter_modules
.. currentmodule:: flax.nnx
.. autoclass:: Module
   :members:
.. autofunction:: view
.. autofunction:: view_info
.. autofunction:: with_attributes