2023-03-20 22:11:26 +00:00
|
|
|
.. include:: ../links.rst
|
|
|
|
|
|
|
|
|
|
##################################
|
|
|
|
|
lightning.fabric.plugins.precision
|
|
|
|
|
##################################
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Precision
|
|
|
|
|
^^^^^^^^^
|
|
|
|
|
|
|
|
|
|
.. TODO(fabric): include DeepSpeedPrecision
|
|
|
|
|
|
|
|
|
|
.. currentmodule:: lightning.fabric.plugins.precision
|
|
|
|
|
|
|
|
|
|
.. autosummary::
|
|
|
|
|
:toctree: ./generated
|
|
|
|
|
:nosignatures:
|
|
|
|
|
:template: classtemplate.rst
|
|
|
|
|
|
|
|
|
|
Precision
|
|
|
|
|
DoublePrecision
|
|
|
|
|
MixedPrecision
|
2023-04-28 16:36:22 +00:00
|
|
|
XLAPrecision
|
|
|
|
|
XLABf16Precision
|
2023-03-20 22:11:26 +00:00
|
|
|
FSDPPrecision
|
