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