variables
------------------------

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

.. autoclass:: BatchStat
   :members:
.. autoclass:: Cache
   :members:
.. autoclass:: Intermediate
   :members:
.. autoclass:: Param
   :members:
.. autoclass:: Variable
   :members:
.. autoclass:: VariableMetadata
   :members:

.. autofunction:: with_metadata

.. autofunction:: variable_name_from_type
.. autofunction:: variable_type_from_name
.. autofunction:: register_variable_name