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