pytorch-lightning

SKILL.md

PyTorch Lightning

Overview

PyTorch Lightning is a deep learning framework that organizes PyTorch code to eliminate boilerplate while maintaining full flexibility. Automate training workflows, multi-device orchestration, and implement best practices for neural network training and scaling across multiple GPUs/TPUs.

When to Use This Skill

This skill should be used when:

Building, training, or deploying neural networks using PyTorch Lightning

Organizing PyTorch code into LightningModules

Configuring Trainers for multi-GPU/TPU training

Implementing data pipelines with LightningDataModules

Working with callbacks, logging, and distributed training strategies (DDP, FSDP, DeepSpeed)

Structuring deep learning projects professionally

Core Capabilities

1. LightningModule - Model Definition