#9 · Primary category: Deep Learning Frameworks

pytorch-lightning

ai artificial-intelligence data-science deep-learning machine-learning python pytorch

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