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. Lightning evolves with you as your projects go from idea to paper/production.

Install Lightning¶

Pip users

pip install pytorch-lightning

Conda users

conda install pytorch-lightning -c conda-forge

Or read the advanced install guide

You can find our 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