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