Table of Contents
Shortcuts

SingleDeviceXLAStrategy

class lightning.fabric.strategies.SingleDeviceXLAStrategy(device, accelerator=None, checkpoint_io=None, precision=None)[source]

Bases: SingleDeviceStrategy

Strategy for training on a single XLA device.

To analyze traffic and optimize your experience, we serve cookies on this site. By clicking or navigating, you agree to allow our usage of cookies. Read PyTorch Lightning's Privacy Policy.