Table of Contents

Shortcuts

TPUPrecision¶

class lightning.fabric.plugins.precision.TPUPrecision[source]¶

Bases: Precision

Precision plugin for TPU integration.

optimizer_step(optimizer, **kwargs)[source]¶

Hook to run the optimizer step.

Return type:: Any