basic\_training
===============

.. automodule:: penzai.deprecated.v1.toolshed.basic_training

   
   
   .. rubric:: Classes

   .. autosummary::
      :toctree: leaf
      :template: pzclass.rst
   
      LossFunction
      TrainState
      TrainStepFunction
   
   

   
   
   .. rubric:: Functions

   .. autosummary::
      :toctree: leaf
      :template: pzbase.rst
   
      build_train_step_fn
      compute_training_outputs_and_updates