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


create_optim_in_bwd_wrapper
===========================

.. autofunction:: create_optim_in_bwd_wrapper