pytorch-lightning - PyTorch Lightning High-Level Training Framework
High-level PyTorch framework providing Trainer class, automatic distributed training, and callbacks system.
Tags
Updated: 2026-09-21Capabilities
Typical Inputs
Typical Outputs
What this skill does
- Train PyTorch models with Trainer
- Execute model validation and testing
- Configure distributed training strategies
- Monitor training using callbacks
- Configure learning rate schedulers
Inputs
- PyTorch model definitions
- DataLoaders for training and testing
- Trainer configuration parameters
Outputs
- Model checkpoint files
- TensorBoard metric logs
- Trained model weights
Requirements
- Python environment with lightning and torch
- CPU, GPU, TPU, or Apple MPS hardware