Lightning-AI / pytorch-lightning
Pretrain, finetune and deploy AI models on multiple GPUs, TPUs with zero code changes.
☆27,963Updated this week
Related projects: ⓘ
- Google Research☆33,807Updated this week
- Visualizer for neural network, deep learning and machine learning models☆27,558Updated this week
- Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more☆29,930Updated this week
- State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and performance on enter…☆13,233Updated last month
- Tensors and Dynamic neural networks in Python with strong GPU acceleration☆82,188Updated this week
- Facebook AI Research Sequence-to-Sequence Toolkit written in Python.☆30,165Updated last week
- 🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.☆132,113Updated this week
- Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70 countries including …☆23,170Updated last month
- The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --…☆31,479Updated this week
- An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model c…☆13,985Updated 2 months ago
- Latex code for making neural networks diagrams☆21,827Updated last year
- Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.☆29,989Updated last week
- Ray is a unified framework for scaling AI and Python applications. Ray consists of a core distributed runtime and a set of AI Libraries f…☆33,048Updated this week
- The fastai deep learning library☆26,108Updated 3 weeks ago
- 🤗 The largest hub of ready-to-use datasets for ML models with fast, easy-to-use and efficient data manipulation tools☆19,012Updated this week
- Open source platform for the machine learning lifecycle☆18,340Updated this week
- A hyperparameter optimization framework☆10,527Updated last week
- A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training☆19,845Updated last month
- A library for efficient similarity search and clustering of dense vectors.☆30,511Updated this week
- PyTorch Tutorial for Deep Learning Researchers☆29,835Updated last year
- Graph Neural Network Library for PyTorch☆20,956Updated this week
- CLIP (Contrastive Language-Image Pretraining), Predict the most relevant text snippet given an image☆24,723Updated last month
- Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125☆14,031Updated this week
- Datasets, Transforms and Models specific to Computer Vision☆15,992Updated this week
- A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.☆22,236Updated this week
- Fast and Accurate ML in 3 Lines of Code☆7,646Updated this week
- Open standard for machine learning interoperability☆17,638Updated this week
- TensorFlow code and pre-trained models for BERT☆37,839Updated last month
- Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities☆19,545Updated 3 weeks ago
- This repository contains implementations and illustrative code to accompany DeepMind publications☆13,071Updated 2 weeks ago