pytorch-lightning

SKILL.md

PyTorch Lightning organizes PyTorch code to eliminate boilerplate while maintaining flexibility.

Benefits: 40+ lines → 15 lines, no device management, automatic distributed

Callbacks: See references/callbacks.md for EarlyStopping, ModelCheckpoint, custom callbacks, and callback hooks.

High-level PyTorch framework with Trainer class, automatic distributed training (DDP/FSDP/DeepSpeed), callbacks system, and minimal boilerplate. Scales from laptop to supercomputer with same code. Use when you want clean training loops with built-in best practices. Source: orchestra-research/ai-research-skills.

View raw