Table of Contents

Shortcuts

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