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
You can find our the list of supported PyTorch versions in our compatibility matrix.
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.