.. role:: hidden :class: hidden-section .. currentmodule:: torchtune.training create_optim_in_bwd_wrapper =========================== .. autofunction:: create_optim_in_bwd_wrapper