build\_train\_step\_fn
======================

.. currentmodule:: penzai.toolshed.basic_training

.. autofunction:: build_train_step_fn