#227 · Primary category: Education & Research

DeepLearningExamples

computer-vision deep-learning drug-discovery forecasting large-language-models mxnet nlp paddlepaddle pytorch recommender-systems speech-recognition speech-synthesis tensorflow tensorflow2 translation

State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and performance on enterprise-grade infrastructure.

Project last updated:08/12/24

GitHub Stars

14.8K

Forks

3.4K

Contributors

114

License

Other

Why we included this project

NVIDIA maintains this collection as its reference set of deep learning examples, and it is better treated as a learning and benchmarking resource than as a single deployable product. Each subdirectory is a self-contained recipe for one model, EfficientNet, Mask R-CNN, BERT, Tacotron, and others, with scripts, Dockerfiles, and Jupyter notebooks that show how to hit published accuracy and throughput on Volta, Turing, and Ampere GPUs. Teams looking for a solid starting point for mixed-precision training, multi-GPU scaling, or TensorRT and Triton inference can adapt these working examples instead of building from scratch. Because the models span computer vision, NLP, speech, and recommender systems across PyTorch, TensorFlow, and MXNet, the repo also works as a survey of how NVIDIA lays out production-grade training pipelines. What makes it worth a look is the curated, reproducible reference implementations and the notebooks that walk through them, not a turnkey application you can run out of the box.

Articles for this project

No articles for this project yet.

To suggest a topic or contribute an article, contact us.

Related projects in this category