API Reference
=============

Binaries
--------

.. toctree::
   :maxdepth: 3

   t5x.train
   t5x.infer
   t5x.eval
   t5x.main

Training
---------

.. toctree::
   :maxdepth: 3

   t5x.trainer   
   t5x.optimizers
   t5x.interactive_model
   t5x.train_state
   t5x.state_utils
   t5x.losses
   t5x.metrics
   t5x.utils
   t5x.adafactor

Inference
---------

.. toctree::
   :maxdepth: 3

   t5x.decoding

Models
------

.. toctree::
   :maxdepth: 3

   t5x.models

Checkpointing
-------------

.. toctree::
   :maxdepth: 3

   t5x.checkpoints
   t5x.checkpoint_utils
   t5x.checkpoint_importer


Paritioning
-----------

.. toctree::
   :maxdepth: 3

   t5x.partitioning

Config
------

.. toctree::
   :maxdepth: 3

   t5x.config_utils
   t5x.gin_utils

Utils
-----

.. toctree::
   :maxdepth: 3

   t5x.test_utils
   t5x.binary_search