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. Lightning evolves with you as your projects go from idea to paper/production.
pip install pytorch-lightning
conda install pytorch-lightning -c conda-forge
Or read the advanced install guide
We are fully compatible with any stable PyTorch version v1.10 and above.
Current Lightning Users¶
Learn Lightning in small bites at 4 levels of expertise: Introductory, intermediate, advanced and expert.
Detailed description of API each package. Assumes you already have basic Lightning knowledge.
From NLP, Computer vision to RL and meta learning - see how to use Lightning in ALL research areas.
Learn how to do everything from hyper-parameters sweeps to cloud training to Pruning and Quantization with Lightning.