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.