.. role:: hidden
    :class: hidden-section
.. currentmodule:: torchtune.utils


get_device
==========

.. autofunction:: get_device