Create Custom AI Models With the NVIDIA TAO Toolkit



NVIDIA TAO Toolkit Version 5.0: What’s New

NVIDIA TAO Toolkit delivers a low-code, open-source AI framework to accelerate vision AI model development suitable for all skill levels—from beginners to expert data scientists. Developers can now use the power and efficiency of transfer learning to achieve state-of-the-art accuracy and production-class throughput in record time with adaptation and optimization.
This latest release of NVIDIA TAO Toolkit 5.0 is leaded with groundbreaking features to boost your AI model development:

  • Export model in standard ONNX format and deploy on any platform—GPUs, CPUs, MCUs, DLAs, etc.
  • State-of-the-art vision Transformers for better accuracy and robustness against image corruption and noise
  • The ability to annotate segmentation masks with new AI-assisted data annotation capabilities
  • Automatic fine-tuning of hyperparameters with AutoML on new vision Transformers
  • New computer vision pre-trained models for optical character detection and recognition, siamese inspection, and more

Download TAO Download from Github Repo Try on LaunchPad

Production-Ready Vision AI

NVIDIA TAO is also available as a part of NVIDIA AI Enterprise, an end-to-end, secure, cloud-native AI software platform optimized to accelerate enterprises to the leading edge of AI.

With enterprise-grade security, stability, manageability, and support, NVIDIA AI Enterprise speeds time to value while mitigating the potential risks of open-source software, ensuring business continuity and a reliable platform for running mission-critical AI applications.

Benefits of using TAO Toolkit with NVIDIA AI Enterprise:


  • Access to exclusive foundation models for vision AI
  • Validation and integration for NVIDIA AI open-source software
  • Access to AI solution workflows to speed time to production
  • Certifications to deploy AI everywhere
  • Enterprise-grade support, security, manageability, and API stability to mitigate potential risks of open source software

Learn More

Get Started With TAO Toolkit on Google Colab

New Developer Blog for 5.0

The next version of TAO Toolkit supercharges vision AI development for practically any developer, in any service and on any device

Read the Blog

New Blog : Vision Transformers

Learn how to improve accuracy and robustness of vision AI apps with Vision Transformers and NVIDIA TAO

Read the Blog

New Blog -Character Detection and Recognition

Learn how to train and deploy a custom optical character detection and recognition model using NVIDIA TAO and NVIDIA Triton.


Read the Blog -
Part 1 |   Part 2


Developer Starter Resources


Training Notebooks & Containers

Sample Deployment Applications

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.

Additional Resources


Product Support

Ethical AI

NVIDIA 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.