• Docs >
  • lightning.fabric.plugins.precision
Shortcuts

lightning.fabric.plugins.precision

Precision

Precision

Base class for all plugins handling the precision-specific parts of the training.

DoublePrecision

Plugin for training with double (torch.float64) precision.

MixedPrecision

Plugin for Automatic Mixed Precision (AMP) training with torch.autocast.

TPUPrecision

Precision plugin for TPU integration.

TPUBf16Precision

Plugin that enables bfloats on TPUs.

FSDPPrecision

AMP for Fully Sharded Data Parallel training.