Start Creating Custom AI Models with the NVIDIA TAO Toolkit

NVIDIA TAO Toolkit Version 3.22.05: What’s New

NVIDIA TAO Toolkit is a low-code AI model development solution that uses the power of transfer learning to simplify and accelerate the creation of custom, production-ready AI models. The new release makes it easy to:

“Bring your own” ONNX models weights into TAO for fine-tuning and optimizing.
Deploy TAO Toolkit as-a-Service in a modern, cloud-native infrastructure on Kubernetes and integrate it with REST APIs.
Visualize the model training progress such as training loss, validation loss, and histogram of model weights in TensorBoard.
Access new vision pretrained models, including Point Cloud, Pose Action Classification, and 3D Pose Estimation.
Take advantage of new speech pretrained models, including HiFi Gan and FastPitch for deploying custom text-to-speech applications.

DOWNLOAD TAO

Get Started With TAO Toolkit

Introductory Whitepaper

Learn how NVIDIA TAO Toolkit and pretrained models can transform your development efforts.

Read Whitepaper

New Developer Blog

Fastrack AI-Powered Robotic Applications with Synthetic Data and Transfer Learning

Read Blog

GTC Webinar

Learn how to create and deploy custom, production-ready vision AI and conversational AI models without any expertise in AI.

Watch Now

LaunchPad

Test Drive the NVIDIA TAO Toolkit for free on NVIDIA LaunchPad.

Sign Up Now

Developer Starter Resources

Vision AI

Developer Blogs	Developing and deploying AI-powered robots with NVIDIA Isaac Sim and NVIDIA TAO Customize Action Recognition with TAO Toolkit and Deploy with DeepStream Training and Optimizing a 2D Pose Estimation Model with the NVIDIA TAO Toolkit, Part 1 \| Part 2 Preparing State-of-the-Art Models for Classification and Object Detection with the NVIDIA TAO Toolkit
Sample Applications	Deploy YoloV3/V4, EfficientDet, UNET, etc. Deploy License Plate Detection and Recognition Deploy 2D Body Pose Estimation, Gaze Estimation, Facial Landmark, HeartRateNet, Gesture, and Emotion To convert TAO Toolkit model (etlt) to an NVIDIA® TensorRT™ engine for deployment with DeepStream, select the appropriate TAO-converter for your hardware and software stack.
Download Resources	For the computer vision models and container collection, Download from NGC For vision AI model, access the collection of Jupyter Notebooks and training specs To deploy TAO Toolkit models using DeepStream, go to download resources . To deploy TAO Toolkit models using NVIDIA Triton™, go to TAO Triton repo

Conversational AI

Developer Blogs	Building and Deploying Conversational AI Models Using the NVIDIA TAO Toolkit Speech Recognition: Customizing Models to Your Domain Using Transfer Learning
Sample Applications	Deploy Conversational AI apps with NVIDIA Riva
Download Resources	For the conversational AI models and container collection, download from NGC. To deploy with Riva, go to download resources Get started with Jupyter Notebooks: Text-to-Speech HiFiGan Text-to-Speech FastPitch Speech Recognition CitriNet Question Answering Text Classification Named Entity Recognition Punctuation & Capitalization Intent Detection & Slot Tagging

Additional Resources

Blogs & Tutorials

Read more developer blogs

Featured Webinars

GTC 2022: AI Model Made Simple
Top 5 Reasons to use TAO Toolkit
GTC 2021: Train Smarter, Not Harder, with NVIDIA
GTC 2021: Building and Deploying a Custom Conversational AI App with NVIDIA TAO Toolkit and Riva

Data Generation And Labeling Partners

NVIDIA has partnered with several companies to bring data creation and annotation to accelerate training.

DLI Training Courses

Documentation

Product Support

TAO Toolkit Forum

NVIDIA AI Enterprise

Ethical AI

NVIDIA’s platforms and application frameworks enable developers to build a wide array of AI applications. Consider potential algorithmic bias when choosing or creating the models being deployed. Also, work with the model’s developer to ensure that it meets the requirements for the relevant industry and use case; that the necessary instruction and documentation are provided to understand error rates, confidence intervals, and results; and that the model is being used under the conditions and in the manner intended.