Welcome to PyTorch Lightning¶

Animation showing how to convert a standard training loop to a Lightning loop

PyTorch Lightning is the deep learning framework for professional AI researchers and machine learning engineers who need maximal flexibility without sacrificing performance at scale.

Install Lightning¶

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

New to Lightning?

Use this 2-step guide to learn key concepts.

PyTorch to PyTorch Lightning

Easily organize your existing PyTorch code into PyTorch Lightning.

Examples

See how Lightning in used in research areas like NLP, Computer Vision, RL and meta learning.

API Reference

Detailed descriptions of each API package.

Common Use Cases

Key Lightning use cases, including hyperparameter sweeps, cloud training, Pruning and Quantization.

Benchmarking

Learn how to benchmark PyTorch Lightning.