TPUBf16PrecisionPlugin¶
- class lightning.pytorch.plugins.precision.TPUBf16PrecisionPlugin(*args, **kwargs)[source]¶
Bases:
lightning.pytorch.plugins.precision.tpu.TPUPrecisionPlugin
Plugin that enables bfloats on TPUs.
Bases: lightning.pytorch.plugins.precision.tpu.TPUPrecisionPlugin
Plugin that enables bfloats on TPUs.