Simplify and Accelerate AI Model Creation
Create highly accurate, customized, and production-ready AI models for speech and vision AI applications with the NVIDIA TAO Toolkit.
Low-Code AI Model Development
The NVIDIA TAO Toolkit, built on TensorFlow and PyTorch, is a low-code version of the NVIDIA Train, Adapt and Optimize (TAO) framework that accelerates model training by abstracting away AI and deep learning framework complexity. The TAO Toolkit lets you use the power of transfer learning to fine-tune NVIDIA pretrained models with your own data and optimize for inference—without AI expertise or large training datasets.
By clicking the above "Agree & Download" button, you are confirming that you have read and agree to be bound by the License For Customer Use of NVIDIA Software
Train Models Easily
Leverage a low-code solution that lets you train models with Jupyter Notebooks, eliminating the need for AI framework expertise.
Build Highly Accurate AI
Use NVIDIA pretrained modelso and 100+ NVIDIA-optimized model architectures to create highly accurate and custom AI models for your use case.
Optimize for Inference
Go beyond customization and achieve up to 4X performance by optimizing the model for inference.
Deploy with Ease
Deploy optimized models using NVIDIA DeepStream for vision AI, NVIDIA® Riva for speech AI, and NVIDIA Triton™ Inference Server.
Key Features
Deploy the TAO Toolkit as a Service with REST APIs
Deploy the TAO Toolkit in a modern cloud-native infrastructure on Kubernetes and integrate it into your application with REST APIs.
Import ONNX Model Weights
Simply import your own classification and segmentation model weights from an ONNX model into TAO Toolkit for fine-tuning and optimization.
Augment Your Data
Improve your model performance with data augmentation features such as spatial and color transformations to diversify your existing image dataset.
Visualize with TensorBoard
Understand how your model converges by visualizing scalars such as training and validation loss, accuracy, model weights, and predicted images.
Optimize for Inference
Achieve up to a 4X speedup in inference with pruning and INT8 quantization while maintaining comparable accuracy.
Customize Pretrained Vision and Speech Models
Develop computer vision applications by customizing pretrained models. Or build high-quality speech services by fine-tuning automatic speech recognition and text-to-speech models, using the TAO Toolkit.
Get Started with the NVIDIA TAO Toolkit
By clicking the above "Agree & Download" button, you are confirming that you have read and agree to be bound by the License For Customer Use of NVIDIA Software