2023-01-10 19:11:03 +00:00
|
|
|
:orphan:
|
|
|
|
|
|
|
|
.. include:: ../../links.rst
|
|
|
|
|
|
|
|
#############
|
|
|
|
API Reference
|
|
|
|
#############
|
|
|
|
|
|
|
|
|
|
|
|
Fabric
|
|
|
|
^^^^^^
|
|
|
|
|
2023-02-27 20:14:23 +00:00
|
|
|
.. currentmodule:: lightning.fabric.fabric
|
2023-01-10 19:11:03 +00:00
|
|
|
|
|
|
|
.. autosummary::
|
|
|
|
:toctree: ../../api
|
|
|
|
:nosignatures:
|
|
|
|
:template: classtemplate.rst
|
|
|
|
|
|
|
|
Fabric
|
|
|
|
|
|
|
|
|
|
|
|
Accelerators
|
|
|
|
^^^^^^^^^^^^
|
|
|
|
|
2023-02-27 20:14:23 +00:00
|
|
|
.. currentmodule:: lightning.fabric.accelerators
|
2023-01-10 19:11:03 +00:00
|
|
|
|
|
|
|
.. autosummary::
|
|
|
|
:toctree: ../../api
|
|
|
|
:nosignatures:
|
|
|
|
:template: classtemplate.rst
|
|
|
|
|
|
|
|
Accelerator
|
|
|
|
CPUAccelerator
|
|
|
|
CUDAAccelerator
|
|
|
|
MPSAccelerator
|
|
|
|
TPUAccelerator
|
|
|
|
|
|
|
|
|
2023-01-13 13:09:44 +00:00
|
|
|
Loggers
|
|
|
|
^^^^^^^
|
|
|
|
|
2023-02-27 20:14:23 +00:00
|
|
|
.. currentmodule:: lightning.fabric.loggers
|
2023-01-13 13:09:44 +00:00
|
|
|
|
|
|
|
.. autosummary::
|
|
|
|
:toctree: ../../api
|
|
|
|
:nosignatures:
|
|
|
|
:template: classtemplate.rst
|
|
|
|
|
|
|
|
Logger
|
|
|
|
CSVLogger
|
|
|
|
TensorBoardLogger
|
|
|
|
|
|
|
|
|
2023-01-10 19:11:03 +00:00
|
|
|
Plugins
|
|
|
|
^^^^^^^
|
|
|
|
|
|
|
|
Precision
|
|
|
|
"""""""""
|
|
|
|
|
|
|
|
.. TODO(fabric): include DeepSpeedPrecision
|
|
|
|
|
2023-02-27 20:14:23 +00:00
|
|
|
.. currentmodule:: lightning.fabric.plugins.precision
|
2023-01-10 19:11:03 +00:00
|
|
|
|
|
|
|
.. autosummary::
|
|
|
|
:toctree: ../../api
|
|
|
|
:nosignatures:
|
|
|
|
:template: classtemplate.rst
|
|
|
|
|
|
|
|
Precision
|
|
|
|
DoublePrecision
|
|
|
|
MixedPrecision
|
|
|
|
TPUPrecision
|
|
|
|
TPUBf16Precision
|
|
|
|
FSDPPrecision
|
|
|
|
|
|
|
|
|
|
|
|
Environments
|
|
|
|
""""""""""""
|
|
|
|
|
2023-02-27 20:14:23 +00:00
|
|
|
.. currentmodule:: lightning.fabric.plugins.environments
|
2023-01-10 19:11:03 +00:00
|
|
|
|
|
|
|
.. autosummary::
|
|
|
|
:toctree: ../../api
|
|
|
|
:nosignatures:
|
|
|
|
:template: classtemplate_noindex.rst
|
|
|
|
|
|
|
|
~cluster_environment.ClusterEnvironment
|
|
|
|
~kubeflow.KubeflowEnvironment
|
|
|
|
~lightning.LightningEnvironment
|
|
|
|
~lsf.LSFEnvironment
|
2023-02-03 10:45:11 +00:00
|
|
|
~mpi.MPIEnvironment
|
2023-01-10 19:11:03 +00:00
|
|
|
~slurm.SLURMEnvironment
|
|
|
|
~torchelastic.TorchElasticEnvironment
|
|
|
|
~xla.XLAEnvironment
|
|
|
|
|
|
|
|
|
|
|
|
IO
|
|
|
|
""
|
|
|
|
|
2023-02-27 20:14:23 +00:00
|
|
|
.. currentmodule:: lightning.fabric.plugins.io
|
2023-01-10 19:11:03 +00:00
|
|
|
|
|
|
|
.. autosummary::
|
|
|
|
:toctree: ../../api
|
|
|
|
:nosignatures:
|
|
|
|
:template: classtemplate.rst
|
|
|
|
|
|
|
|
~checkpoint_io.CheckpointIO
|
|
|
|
~torch_io.TorchCheckpointIO
|
|
|
|
~xla.XLACheckpointIO
|
|
|
|
|
|
|
|
|
|
|
|
Collectives
|
|
|
|
"""""""""""
|
|
|
|
|
2023-02-27 20:14:23 +00:00
|
|
|
.. currentmodule:: lightning.fabric.plugins.collectives
|
2023-01-10 19:11:03 +00:00
|
|
|
|
|
|
|
.. autosummary::
|
|
|
|
:toctree: ../../api
|
|
|
|
:nosignatures:
|
|
|
|
:template: classtemplate.rst
|
|
|
|
|
|
|
|
Collective
|
|
|
|
TorchCollective
|
|
|
|
SingleDeviceCollective
|
|
|
|
|
|
|
|
|
|
|
|
Strategies
|
|
|
|
^^^^^^^^^^
|
|
|
|
|
|
|
|
.. TODO(fabric): include DeepSpeedStrategy, XLAStrategy
|
|
|
|
|
2023-02-27 20:14:23 +00:00
|
|
|
.. currentmodule:: lightning.fabric.strategies
|
2023-01-10 19:11:03 +00:00
|
|
|
|
|
|
|
.. autosummary::
|
|
|
|
:toctree: ../../api
|
|
|
|
:nosignatures:
|
|
|
|
:template: classtemplate.rst
|
|
|
|
|
|
|
|
Strategy
|
|
|
|
DDPStrategy
|
|
|
|
DataParallelStrategy
|
|
|
|
FSDPStrategy
|
|
|
|
ParallelStrategy
|
|
|
|
SingleDeviceStrategy
|
|
|
|
SingleTPUStrategy
|