===============
fairseq2.device
===============

.. automodule:: fairseq2.device
    :no-members:

Interfaces
==========

.. autoclass:: CudaContext
.. autoclass:: SupportsDeviceTransfer

Classes
=======

.. autoclass:: StandardCudaContext

Functions
=========

.. autofunction:: get_default_device
.. autofunction:: get_current_device

Exceptions
==========

.. autoclass:: LocalRankOutOfRangeError