NVIDIA / DeepLearningExamples
State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and performance on enterprise-grade infrastructure.
☆13,233Updated last month
Related projects: ⓘ
- Pretrain, finetune and deploy AI models on multiple GPUs, TPUs with zero code changes.☆27,963Updated this week
- The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --…☆31,479Updated this week
- Google Research☆33,807Updated this week
- A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.☆22,236Updated this week
- Visualizer for neural network, deep learning and machine learning models☆27,558Updated this week
- Datasets, Transforms and Models specific to Computer Vision☆15,992Updated this week
- Library of deep learning models and datasets designed to make deep learning more accessible and accelerate ML research.☆15,324Updated last year
- Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125☆14,031Updated this week
- PyTorch tutorials.☆8,102Updated 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
- PyTorch Tutorial for Deep Learning Researchers☆29,835Updated last year
- A comprehensive list of pytorch related content on github,such as different models,implementations,helper libraries,tutorials etc.☆15,313Updated 7 months ago
- An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model c…☆13,985Updated 2 months ago
- Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.☆14,158Updated 2 weeks ago
- Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.☆29,989Updated last week
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,303Updated 2 weeks ago
- The fastai deep learning library☆26,108Updated 3 weeks ago
- The Incredible PyTorch: a curated list of tutorials, papers, projects, communities and more relating to PyTorch.☆11,366Updated 3 months ago
- Facebook AI Research Sequence-to-Sequence Toolkit written in Python.☆30,165Updated last week
- A library for efficient similarity search and clustering of dense vectors.☆30,511Updated this week
- Latex code for making neural networks diagrams☆21,827Updated last year
- Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more☆29,930Updated this week
- Open standard for machine learning interoperability☆17,638Updated this week
- End-to-End Object Detection with Transformers☆13,367Updated 6 months ago
- Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Py…☆19,651Updated 3 weeks ago
- TensorFlow code and pre-trained models for BERT☆37,839Updated last month
- A collection of various deep learning architectures, models, and tips☆16,585Updated 7 months ago
- A playbook for systematically maximizing the performance of deep learning models.☆26,385Updated 3 months ago
- Graph Neural Network Library for PyTorch☆20,956Updated this week
- Trax — Deep Learning with Clear Code and Speed☆8,044Updated last week