.. role:: hidden :class: hidden-section .. currentmodule:: torchtune.utils get_device ========== .. autofunction:: get_device