lightning.fabric.plugins.precision¶
Precision¶
Base class for all plugins handling the precision-specific parts of the training. |
|
Plugin for training with double ( |
|
Plugin for Automatic Mixed Precision (AMP) training with |
|
Precision plugin for TPU integration. |
|
Plugin that enables bfloats on TPUs. |
|
AMP for Fully Sharded Data Parallel training. |