.. role:: hidden
    :class: hidden-section
.. currentmodule:: torchtune.training


register_optim_in_bwd_hooks
===========================

.. autofunction:: register_optim_in_bwd_hooks