NVIDIA TAO
NVIDIA Train, Adapt, and Optimize (TAO) is an AI-model-adaptation platform that simplifies and
accelerates the creation of enterprise AI applications and services. By fine-tuning pretrained models
with custom data through a UI-based, guided workflow, enterprises can produce highly accurate models in
hours rather than months, eliminating the need for large training runs and deep AI expertise.
Getting Started
TensorFlow
ContainerTensorFlow is an open source platform for machine learning. It provides comprehensive tools and libraries in a flexible architecture allowing easy deployment across a variety of platforms and devices.
PyTorch
ContainerPyTorch is a GPU accelerated tensor computational framework. Functionality can be extended with common Python libraries such as NumPy and SciPy. Automatic differentiation is done with a tape-based system at the functional and neural network layer levels.
TAO Toolkit - Computer Vision
Collection - Deep LearningTAO Toolkit is a python based AI toolkit for taking purpose-built pre-trained AI models and customizing them with your own data.
NGC - Getting Started
Collection - BeginnerLooking to get started with containers and models on NGC? This is the place to start.
Documentation
We've got a whole host of documentation, covering the NGC UI and our powerful CLI. You can find out more here. Go to Documentation
Command Line Interface
Want to get more from NGC? Everything you see here can be used and managed via our powerful CLI tools. Download Now
NGC Private Registry
Private Registries from NGC allow you to secure, manage, and deploy your own assets to accelerate your journey to AI. Learn More
Popular CollectionsSee All Collections
Build AI on AWS
Collection - InfrastructureAWS Collection. From SageMaker to P3 instances in the cloud, whether you're working with PaaS or infrastructure this collection is the place to start with NGC and Amazon
Automatic Speech Recognition
Collection - Automatic Speech RecognitionA collection of easy to use, highly optimized Deep Learning Models for Recommender Systems. Deep Learning Examples provides Data Scientist and Software Engineers with recipes to Train, fine-tune, and deploy State-of-the-Art Models
Clara Discovery
Collection - HealthcareClara Discovery is a collection of frameworks, applications, and AI models enabling GPU-accelerated computational drug discovery
Clara NLP
Collection - HealthcareClara NLP is a collection of SOTA biomedical pre-trained language models as well as highly optimized pipelines for training NLP models on biomedical and clinical text
Clara Train
Collection - HealthcareClara Train - domain optimized training framework - includes Clara Train container, models, getting started jupyter notebook, utilitie
CUDA Toolkit
Collection - High Performance ComputingCUDA Toolkit provides the core, foundational development environment for creating high performance NVIDIA GPU-accelerated applications for diverse workloads from high performance computing, data science analytics and AI.
DeepStream - CV Deployment
Collection - Intelligent Video AnalyticsDeepStream SDK delivers a complete streaming analytics toolkit for AI based video and image understanding and multi-sensor processing. The DeepStream SDK brings deep neural networks and other complex processing tasks into a stream processing pipeline.
NGC - Getting Started
Collection - BeginnerLooking to get started with containers and models on NGC? This is the place to start.
Popular ContainersSee All Containers
Validator for NVIDIA GPU Operator
ContainerValidates NVIDIA GPU Operator components
NVIDIA Driver Manager For Kubernetes
ContainerManages NVIDIA Driver upgrades in Kubernetes cluster.
CUDA
ContainerCUDA is a parallel computing platform and programming model that enables dramatic increases in computing performance by harnessing the power of the NVIDIA GPUs.
DeepStream
ContainerDeepStream SDK delivers a complete streaming analytics toolkit for AI based video and image understanding and multi-sensor processing. This container is for data center GPUs such as NVIDIA T4 running on x86 platform.
Popular ModelsSee All Models
clara_pt_covid19_ct_lesion_segmentation
ModelA pre-trained model for volumetric (3D) segmentation of the COVID-19 lesion from CT images.
clara_pt_fed_learning_brain_tumor_mri_segmentation
ModelA federated learning demo for volumetric (3D) segmentation of brain tumors from multimodal MRIs.
clara_pt_pathology_metastasis_detection
ModelA pre-trained model for automated detection of metastases in whole-slide histopathology images.
clara_pt_prostate_mri_segmentation
ModelA pre-trained model for volumetric (3D) segmentation of the prostate central gland and peripheral zone from the multimodal MR (T2, ADC).
Popular ResourcesSee All Resources
Fine-Tuning COVID-19 CT Scan Classifier
ResourceExample Jupyter Notebook Sample Code to Fine-Tune Clara Train COVID-19 CT Scan Classification Pretrained Model with Custom Dataset
BERT for PyTorch
ResourceBERT is a method of pre-training language representations which obtains state-of-the-art results on a wide array of NLP tasks.
BERT for TensorFlow
ResourceBERT is a method of pre-training language representations which obtains state-of-the-art results on a wide array of NLP tasks.
BERT for TensorFlow Jupyter Notebook
ResourceJupyter Notebook example for Question Answering with BERT for TensorFlow
Popular Helm ChartsSee All Helm Charts
NVIDIA Network Operator
Helm ChartNvidia Network Operator Helm Chart provides an easy way to install, configure and manage the lifecycle of Nvidia Mellanox network operator.
NVSM
Helm ChartA Helm chart for deploying Nvidia System Management software on DGX Nodes
GPU Operator
Helm ChartDeploy and Manage NVIDIA GPU resources in Kubernetes.
NVIDIA Device Plugin for Kubernetes
Helm ChartExpose NVIDIA GPUs as Resources in Kubernetes