pytorch-lightning
AI model builder
A framework for building and training AI models on multiple GPUs or TPUs with minimal code changes.
Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.
28k stars
250 watching
3k forks
Language: Python
last commit: 3 days ago
Linked from 5 awesome lists
aiartificial-intelligencedata-sciencedeep-learningmachine-learningpythonpytorch
vinta/awesome-python
josephmisiti/awesome-machine-learning
ethicalml/awesome-production-machine-learning
krzjoa/awesome-python-data-science
se-ml/awesome-semlRelated projects:
| Repository | Description | Stars |
|---|---|---|
 lightning-universe/lightning-bolts | Provides a toolbox of components to extend PyTorch Lightning for deep learning research and production | 1,693 |
 pytorch/pytorch | A Python library providing tensors and dynamic neural networks with strong GPU acceleration | 83,959 |
 sktime/pytorch-forecasting | A PyTorch-based package for state-of-the-art time series forecasting with deep learning architectures | 4,001 |
| lightning-ai/deep-learning-project-template | Provides a standard template for structuring and organizing deep learning projects | 1,242 |
 ahkarami/deep-learning-in-production | A collection of notes and references on deploying deep learning models in production environments | 4,306 |
| pytorch/ignite | A high-level library to help with training and evaluating neural networks in PyTorch | 4,526 |
 qubvel-org/segmentation_models.pytorch | A PyTorch library for building and training neural networks for image segmentation tasks. | 9,696 |
 yunjey/pytorch-tutorial | A comprehensive tutorial project that provides code examples for learning PyTorch by implementing various deep learning models and demonstrating their usage. | 30,251 |
 cadene/pretrained-models.pytorch | Provides pre-trained PyTorch models for various computer vision tasks and datasets. | 9,036 |
| lightning-ai/lit-llama | An implementation of a large language model using the nanoGPT architecture | 5,993 |
 pyg-team/pytorch_geometric | A PyTorch-based library for training and applying Graph Neural Networks to structured data | 21,382 |
 luolc/adabound | An optimizer that combines the benefits of Adam and SGD algorithms | 2,907 |
 jfzhang95/pytorch-deeplab-xception | A PyTorch implementation of the DeepLab-V3-Plus model with support for multiple backbones and datasets | 2,910 |
 catalyst-team/catalyst | A PyTorch framework for accelerating deep learning research and development by focusing on reproducibility, rapid experimentation, and codebase reuse. | 3,295 |
 ludwig-ai/ludwig | A low-code framework for building custom deep learning models and neural networks | 11,189 |