Software Development
PyTorch Lightning
Organizes PyTorch training code with the lightning package (PyTorch Lightning): LightningModule, LightningDataModule, Trainer, callbacks such as ModelCheckpoint and EarlyStopping, loggers (TensorBoard, W&B, MLflow, Comet, CSV), and multi-GPU/TPU strategies (DDP, FSDP, DeepSpeed). Use when structuring a PyTorch model into training, validation and test steps. Use when configuring a Trainer for multi-GPU or TPU runs. Use when writing a LightningDataModule for data loading. Use when adding checkpointing, early stopping or experiment logging to a training run. Use when choosing between DDP, FSDP and DeepSpeed for a model size. Not for plain PyTorch loops without Lightning.
- Publisher / source
- KalarisLabs/research-agent-skills (found on the CSV list)
- Upstream record
- KalarisLabs/research-agent-skills
- Version
- Version 2
- Published
- Publication review
- Approved for publication
- Content fingerprint
8d6e1a14993fe8bc7240ce4193707cd193c0eb3aca26c44a583859b8fb52c6cb