Lightning-AI / pytorch-lightningLinks
Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.
ā30,194Updated last week
Alternatives and similar repositories for pytorch-lightning
Users that are interested in pytorch-lightning are comparing it to the libraries listed below
Sorting:
- š Geometric Computer Vision Library for Spatial AIā10,782Updated last week
- š A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (iā¦ā9,180Updated last week
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)ā9,205Updated last month
- Tensors and Dynamic neural networks in Python with strong GPU accelerationā93,636Updated this week
- A hyperparameter optimization frameworkā12,789Updated this week
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorchā8,813Updated this week
- High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.ā4,701Updated last week
- A scikit-learn compatible neural network library that wraps PyTorchā6,115Updated last month
- Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and moreā33,605Updated this week
- Model interpretability and understanding for PyTorchā5,419Updated this week
- Google Researchā36,469Updated this week
- Serve, optimize and scale PyTorch models in productionā4,350Updated 2 months ago
- Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125ā15,161Updated 3 months ago
- Python package built to ease deep learning on graph, on top of existing DL frameworks.ā14,086Updated 2 months ago
- Graph Neural Network Library for PyTorchā22,954Updated this week
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)ā7,964Updated last month
- Open standard for machine learning interoperabilityā19,665Updated last week
- A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.ā23,439Updated last month
- PyTorch3D is FAIR's library of reusable components for deep learning with 3D dataā9,534Updated 3 weeks ago
- Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalitiesā21,753Updated 3 months ago
- The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --ā¦ā35,398Updated this week
- DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.ā40,264Updated last week
- Visualizer for neural network, deep learning and machine learning modelsā31,501Updated this week
- Ongoing research training transformer models at scaleā13,755Updated this week
- Facebook AI Research Sequence-to-Sequence Toolkit written in Python.ā31,856Updated last week
- Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.ā14,601Updated 2 weeks ago
- Label Studio is a multi-type data labeling and annotation tool with standardized output formatā24,927Updated this week
- Hydra is a framework for elegantly configuring complex applicationsā9,792Updated last month
- The open source developer platform to build AI/LLM applications and models with confidence. Enhance your AI applications with end-to-end ā¦ā22,293Updated last week
- Flax is a neural network library for JAX that is designed for flexibility.ā6,845Updated this week