TPUBf16PrecisionPlugin

class pytorch_lightning.plugins.precision.TPUBf16PrecisionPlugin[source]

Bases: pytorch_lightning.plugins.precision.tpu.TPUPrecisionPlugin

Plugin that enables bfloats on TPUs.

connect(model, optimizers, lr_schedulers)[source]

Connects this plugin to the accelerator and the training process.

Return type

Tuple[Module, List[Optimizer], List[Any]]