pytorch-lightning - PyTorch Lightning Training Framework
High-level PyTorch framework for distributed training with automatic DDP/FSDP/DeepSpeed support
Tags
Updated: 2026-05-28Capabilities
Typical Inputs
Typical Outputs
What this skill does
- train model
- validate model
- test model
- log metrics
- save checkpoint
- adjust learning rate
- configure optimizer
- distribute training
- switch devices
- monitor training
Inputs
- PyTorch model
- DataLoader
- training configuration
Outputs
- trained model
- model checkpoint
- training log
- training metrics
Requirements
- lightning library
- torch
- transformers
- CPU/GPU/TPU/MPS support
- FP32/FP16/BF16/FP8 precision
