maxtext.layers.train\_state\_nnx module
=======================================

.. automodule:: maxtext.layers.train_state_nnx
   :members:
   :show-inheritance:
   :undoc-members: