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.

Install Lightning

Pip users

pip install lightning

Conda users

conda install lightning -c conda-forge

Or read the advanced install guide

You can find the list of supported PyTorch versions in our compatibility matrix.

Get Started

• Lightning in 15 minutes — Learn the 7 key steps of a typical Lightning workflow.

• Benchmarking — Learn how to benchmark PyTorch Lightning.

Current Lightning Users

• Level Up! — Learn Lightning in small bites at 4 levels of expertise: Introductory, intermediate, advanced and expert.

• API Reference — Detailed description of API each package. Assumes you already have basic Lightning knowledge.

• Hands-on Examples — From NLP, Computer vision to RL and meta learning - see how to use Lightning in ALL research areas.

• Common Workflows — Learn how to do everything from hyper-parameters sweeps to cloud training to Pruning and Quantization with Lightning.

• Convert code to PyTorch Lightning — Convert your current code to Lightning