Welcome to PyTorch Lightning¶
PyTorch Lightning is the deep learning framework for professional AI researchers and machine learning engineers who need maximal flexibility without sacrificing performance at scale.
For pip (and conda) users
pip install pytorch-lightning
Or directly from conda
conda install pytorch-lightning -c conda-forge
Or read the advanced install guide
See how Lightning in used in research areas like NLP, Computer Vision, RL and meta learning.
Common Use Cases
Key Lightning use cases, including hyperparameter sweeps, cloud training, Pruning and Quantization.