Jupyter Notebooks and Model Scripts | NVIDIA NGC

The NGC catalog offers step-by-step instructions and scripts for creating deep learning models, with sample performance and accuracy metrics to compare your results. These scripts provide expert guidance on building DL models for image classification, language translation, text-to-speech and more. Data scientists can quickly build performance-optimized models by easily adjusting the hyperparameters.

Sort: Last Modified

Colonoscopy Sample App Data

Resource

Clara Holoscan Sample App Data for AI Colonoscopy Segmentation of Polyps

Endoscopy Sample App Data

Resource

Clara Holoscan Sample App Data for AI-based Endoscopy Tool Tracking

BERT PaddlePaddle codebase

Resource

PaddlePaddle codebase for training and using BERT model

BERT for PaddlePaddle

Resource

BERT is a method of pre-training language representations which obtains state-of-the-art results on a wide array of NLP tasks.

Resource

NGC CLI

Multi-AI Sample App Data

Resource

Clara Holoscan Sample App Data for Multi-AI Ultrasound Pipeline

SSD PyTorch codebase

Resource

PyTorch codebase for training and using SSD model

nnUNet TensorFlow2 codebase

Resource

TensorFlow2 codebase for training and using nnUNet model

MaskRCNN PyTorch codebase

Resource

PyTorch codebase for training and using MaskRCNN model

BERT PyTorch codebase

Resource

PyTorch codebase for training and using BERT model

Merlin TensorFlow Jupyter Notebooks

ResourceQuick Deploy

This resource is a collection of Jupyter notebook examples to provide training example for NVIDIA Merlin using Tensorflow.

Merlin PyTorch Jupyter Notebooks

ResourceQuick Deploy

This resource is a collection of Jupyter notebook examples to provide training example for NVIDIA Merlin using Pytorch.

Merlin HugeCTR Jupyter Notebooks

ResourceQuick Deploy

This resource is a collection of Jupyter notebook examples to provide training example for NVIDIA Merlin.

Training with WarpDrive

Resource

Training with WarpDrive

End-to-end Training with WarpDrive

Resource

Jupyter notebook for end-to-end training with WarpDrive

WarpDrive Resetter

Resource

Tutorial covering the basics of WarpDrive Reset and Log

WarpDrive Sampler for Numba

Resource

Tutorial covering the basics of WarpDrive Sampler using Numba

WarpDrive Sampler for PyCUDA

Resource

Tutorial covering the basics of WarpDrive Sampler using PyCUDA

WarpDrive Basics for Numba

Resource

Tutorial covering the basics of WarpDrive using Numba

WarpDrive Basics for PyCUDA

Resource

Tutorial covering the basics of WarpDrive using PyCUDA

DOCA Container Resources

Resource

Set of container configuration files for the various DOCA containers.

Wide & Deep for TensorFlow2

Resource

Wide & Deep Recommender model.

Wide & Deep for TensorFlow1

Resource

Wide & Deep Recommender model.

VAE for TensorFlow1

Resource

The Variational Autoencoder for collaborative filtering focuses on providing recommendations. This is an optimized implementation.

3D-UNet Medical Image Segmentation for TensorFlow1

Resource

A convolutional neural network for 3D image segmentation.