PyTorch Lightning

The ray.train.lightning module adapts a PyTorch Lightning Trainer to run on Ray Train.

prepare_trainer	Prepare the PyTorch Lightning Trainer for distributed execution.
RayLightningEnvironment	Setup Lightning DDP training environment for Ray cluster.
RayDDPStrategy	Subclass of DDPStrategy to ensure compatibility with Ray orchestration.
RayFSDPStrategy	Subclass of FSDPStrategy to ensure compatibility with Ray orchestration.
RayDeepSpeedStrategy	Subclass of DeepSpeedStrategy to ensure compatibility with Ray orchestration.
RayTrainReportCallback	A simple callback that reports checkpoints to Ray on train epoch end.