lightning/docs/source-fabric/api_reference.rst

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

141 lines
2.3 KiB
ReStructuredText
Raw Normal View History

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