pytorch-lightning
by Lightning-AI
Pythonpushed almost 2 years ago
Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.
AI summary
AI model builder
A framework for building and training AI models on multiple GPUs or TPUs with minimal code changes.
- stars
- 28.6K
- forks
- 3.4K
- watching
- 252
- awesome lists
- 5
Featured in 5 awesome lists
Each link jumps to the spot where the list mentions pytorch-lightning.