$ account Lightning-AI
PyTorch Lightning is a deep learning framework that provides a high-level PyTorch training interface (Trainer, LightningModule, LightningDataModule) and a lower-level Fabric API for distributed training. It is a library, not a standalone application. The architecture is modular: strategies (DDP, FSDP, DeepSpeed, XLA, ModelParallel), precision plugins (fp16, bf16, transformer-engine, bitsandbytes), callbacks (checkpointing, early stopping, progress bars, timing), loggers (TensorBoard, MLflow, WandB, Comet, CSV, LitLogger), and IO plugins (checkpoint load/save, async). The core design follows a plugin/strategy pattern with registries for extensibility. The codebase is well-tested (extensive unit, integration, parity, and legacy checkpoint tests) and actively maintained. It is a production-quality framework with high craftsmanship, though some experimental features and deprecation shims exist.