t5x.train_state package
========================

.. currentmodule:: t5x.train_state

.. automodule:: t5x.train_state
  :members: