TPUPrecision

class lightning_fabric.plugins.precision.TPUPrecision[source]

Bases: lightning_fabric.plugins.precision.precision.Precision

Precision plugin for TPU integration.

optimizer_step(optimizer, **kwargs)[source]

Hook to run the optimizer step.

Return type

Any