#9 · Primary category: Deep Learning Frameworks
pytorch-lightning
Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.
Project last updated:08/25/26
GitHub Stars
31.3K
Forks
3.8K
Contributors
1.1K
License
Apache-2.0
Why we included this project
Most PyTorch projects eventually hit the same wall: the training loop works on one GPU, then you need mixed precision, checkpointing, and multi-node orchestration, and suddenly you are maintaining infrastructure instead of models. PyTorch Lightning wraps your code in a LightningModule and Trainer so that scaling from a single CPU to a cluster of GPUs does not require rewriting the core logic. Researchers and engineers get reproducible experiment structure out of the box, and the companion Fabric library offers a lower-level path for those who want finer control without the full framework. It is a solid pick for anyone building custom models in PyTorch who would rather spend time on architecture than on plumbing.
Articles for this project
No articles for this project yet.
To suggest a topic or contribute an article, contact us.
Related projects in this category
tensorflow
An Open Source Machine Learning Framework for Everyone
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
keras
Deep Learning for humans
nanoGPT
The simplest, fastest repository for training/finetuning medium-sized GPTs.
ray
Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.