Dev & Engineering ✓ NVIDIA · Official

TAO Hugging Face Model Integrator

Connect Hugging Face computer-vision models to NVIDIA TAO training, ONNX export, and TensorRT deployment workflows.

51/ 100
Use with care

Useful, but reliability, evidence or controls still have material gaps.

See how it was scored ↓
Works as-is in
Claude Code · Codex(Partial support)
Stars
★ 3.8k
Last updated
today
License
Apache-2.0
huggingfacenvidia-taocomputer-visiontensorrt
+2onnx-exportdocker-validation

What does this skill do, and when should you use it?

This skill is for developers who want to turn a Hugging Face computer-vision model into a TAO Toolkit model. It coordinates a seven-phase workflow covering environment checks, model validation, codebase research, tao-core configuration, tao-pytorch training, ONNX export, and tao-deploy TensorRT integration. It supports classification, object detection, semantic, instance and panoptic segmentation, zero-shot detection, and depth estimation. All changes and validation are intended to run locally inside Docker, with an iterative build, test, debug, fix, and retest loop.

Checks Python, Git, NVIDIA drivers, CUDA, Docker, NVIDIA Container Toolkit, NGC authentication, and Hugging Face credentials; prepares TAO Toolkit container images; inspects the Hugging Face task type, configuration, weight schema, and ONNX export; creates tao-core experiment configuration; implements training, inference, evaluation, and MLOps in tao-pytorch; adds ONNX export and TensorRT engine, inference, and evaluation support in tao-deploy; registers command-line entry points and L0 tests; runs containerized unit tests, static checks, wheel builds, and end-to-end train→export→TensorRT→inference/evaluate validation; and optionally tunes accuracy, latency, or model size.

Good fit
  • A computer-vision developer wants to integrate a Hugging Face ViT, DETR, or SegFormer model with TAO Toolkit.
  • A TAO user needs native training, evaluation, inference, and configuration support for a new Hugging Face model.
  • A deployment engineer needs to export a Hugging Face model to ONNX and build a TensorRT inference path.
  • A team wants to validate training artifacts and TensorRT outputs on a local NVIDIA GPU Docker setup.

How do you install this skill?

Before you use it
  • The skill requires NGC and HuggingFace access plus an NVIDIA GPU, CUDA 13, and Docker; core execution may be unavailable on mainland-China networks or without the required credentials.
  • Handling guidance for HF_TOKEN and NGC credentials is insufficient; confirm that secrets will not enter logs, shell history, or persistent caches before execution.
  • Phase 0 performs local state changes including image pulls, package removal, and scratch cleanup; a general rollback procedure is not provided.
  • The benchmark report contains only one evaluation task and does not establish reliable full-model porting, TensorRT deployment, or end-to-end results.
Before you start
Your agent needs
  • Shell / CLI
  • Network access
  • Local filesystem
Install first
  • Python 3.10+
  • Git
  • NVIDIA driver
  • CUDA 13.0+
  • Docker
  • nvidia-container-toolkit
  • NGC API key
  • HF_TOKEN
  • TAO Toolkit container images
  • Local clones of tao-core, tao-pytorch, tao-deploy, and tao-dataservices

Install it from the NVIDIA/skills collection with the skills CLI: npx skills add nvidia/skills --skill tao-port-huggingface-model --yes. The README does not document the exact installation directory; the CLI prompts for the destination. The skill is located at skills/tao-port-huggingface-model/SKILL.md.

Generic route: install into Claude Code manually (macOS / Linux)
tmp="$(mktemp -d)"
git clone --depth 1 https://github.com/NVIDIA/skills.git "$tmp"
mkdir -p ~/.claude/skills
cp -R "$tmp/skills/tao-port-huggingface-model" ~/.claude/skills/
rm -rf "$tmp"

Generated from the source repository and skill path; it copies only this skill's folder. If the author's install steps above differ, follow those first. To scope it to one project, replace ~/.claude/skills with that project's .claude/skills.

How do you use this skill?

Try saying

Once installed, send your agent any of these to trigger it:

  • Integrate org/model-name into TAO Toolkit.

Give the agent a Hugging Face model ID or URL and request TAO integration, for example: “Integrate org/model-name into TAO Toolkit.” Before starting, prepare NGC authentication, HF_TOKEN, a model short name, and exact image references for tao-pytorch, tao-deploy, and optionally tao-dataservices; Phase 0 asks for these inputs.

What are this skill's strengths and limitations?

Pros
  • Covers the full path from model inspection through TensorRT deployment and containerized end-to-end validation.
  • Requires continuous testing and repair after failures instead of accumulating untested code.
  • Addresses multiple computer-vision tasks and provides a TAO reference-model mapping approach.
  • Explicitly keeps edits, builds, and tests local.
Limitations
  • The setup is demanding: it requires NVIDIA GPU software, Docker, NGC credentials, HF_TOKEN, and TAO container images.
  • The workflow requires local clones of four TAO repositories and careful handling of the tao-core submodule override.
  • The supplied source provides no actual test results, performance measurements, or platform coverage for this specific skill.
  • Execution depends on several referenced documents that are not included in the supplied material.
  • The workflow does not publish code, images, or artifacts to remote repositories or registries.

How does this skill compare with similar options?

Side by side with related skills; every score comes from the same FSRS standard.

Skill FS score Stars Last updated License
TAO Hugging Face Model Integrator this page ✓ NVIDIA · Official 51 · Use with care ★ 3.8k today Apache-2.0
DeepStream Vision Model Importer ✓ NVIDIA · Official 48 · Use with care ★ 3.8k today Apache-2.0
Azure AI Vision Image Analysis ✓ Microsoft · Official 55 · Use with care ★ 2.7k 3mo ago MIT
NVIDIA DeepStream Development ✓ NVIDIA · Official 49 · Use with care ★ 3.8k today Apache-2.0
TAO OCRNet Scene Text Recognition ✓ NVIDIA · Official 48 · Use with care ★ 3.8k today Apache-2.0

How did FollowSkills review this skill?

FollowSkills review · FSRS-2.0
Use with care
51/ 100 5-point scale 2.6 / 5
The upstream repository has new commits since this review. The score still applies to the reviewed revision shown and may not cover the latest changes.
1Trust17 / 25 · 3.4/5

The skill explicitly enforces local-only changes, forbids remote pushes, and requests user-supplied images, credentials, and confirmations at key gates. However, it requires HF_TOKEN, an NGC API key, Docker/GPU privileges, and write access across multiple repositories; credentials are passed through environment variables, while fine-grained permissions, sensitive-data handling, rollback, and complete data-flow disclosure are incomplete, so points are deducted.

2Reliability8 / 20 · 2.0/5

The documentation provides phase gates, a build-test-fix loop, container isolation, and a troubleshooting table, making the main path statically coherent. However, many critical steps depend on unprovided reference files, external TAO repositories, and specific images, and there is no committed executable evidence covering this skill's key paths; the static ceiling therefore limits the score.

3Adaptability9 / 15 · 3.0/5

Triggers, supported computer-vision tasks, rejected non-vision tasks, and phase goals are clearly described. The environment is nevertheless highly demanding, depending on CUDA 13, NVIDIA container tooling, NGC images, and HuggingFace access, with no mainland-China reachability assessment, alternative platform, or sufficiently explicit non-fit boundaries; points are deducted.

4Convention8 / 15 · 2.7/5

SKILL.md has useful phase organization, cross-references, version metadata, an Apache-2.0 declaration, author attribution, and publisher ownership. It lacks the recommended Instructions and Examples sections, has incomplete changelog and maintenance-update guidance, uses a nonstandard author format, and distributes important assumptions across many external references, so points are deducted.

5Effectiveness6 / 15 · 2.0/5

The skill defines an end-to-end objective spanning HuggingFace inspection, TAO training, ONNX, TensorRT, packaging, and container validation, with artifact flow and gates. However, representative final outputs are not statically verifiable, the workflow is expensive, and model adaptation still requires substantial manual implementation. The limited benchmark report supports only plausible core-task completion, so the score remains below the static cap.

6Verifiability3 / 10 · 1.5/5

The files contain concrete commands, checklists, artifact paths, and a limited evaluation report. However, the evaluation covers only one planning task and there is no committed test suite covering the key implementation paths or multi-source corroboration; conclusions rely mainly on author-provided material, so only limited credit is justified.

1 2 3 4 5 6

Open a dimension to read why it scored that way

Reviewed Jul 20, 2026 Reviewed revision 55f18499943e Review evidence[1][2][3][4][5][6][7][8][9][10][11]

Evidence confidence:Low — Mostly static review, author material or a limited demo; useful for discovery, not high-risk decisions.

See the full review method →

FAQ

Is this suitable for an environment without an NVIDIA GPU?
Not for the documented default workflow. It requires an NVIDIA driver, CUDA 13.0+, Docker, nvidia-container-toolkit, and accessible TAO container images.
Can it process any Hugging Face model?
No. Phase 1 rejects non-computer-vision tasks such as NLP, audio, or LLM pipelines, and stops when configuration loading fails or ONNX export cannot work without a rewrite path.
Will it publish code or Docker images remotely?
No. The skill requires remote resources to be read or cloned only; edits, builds, and tests remain local.

More skills from this repository

All from NVIDIA/skills

Data & Analysis ✓ NVIDIA · Official

TAO Depth Anything v2

Train, evaluate, export, and deploy monocular depth models from single RGB images with NVIDIA TAO.

★ 3.8k FS 50 Use with care today
Data & Analysis ✓ NVIDIA · Official

TAO Grounding DINO Open-Vocabulary Detection

Train and deploy text-guided object detectors without a fixed class vocabulary.

★ 3.8k FS 56 Use with care today
Data & Analysis ✓ NVIDIA · Official

TAO SegFormer Semantic Segmentation

Guidance for training, evaluating, exporting, and deploying TAO SegFormer models.

★ 3.8k FS 52 Use with care today
Dev & Engineering ✓ NVIDIA · Official

DeepStream Vision Model Importer

Automates object-detection model import, benchmarking, and reporting for NVIDIA DeepStream.

★ 3.8k FS 48 Use with care today
Data & Analysis ✓ NVIDIA · Official

Hugging Face GPU Fine-Tuning

Build reproducible Hugging Face fine-tuning pipelines on local NVIDIA GPUs.

★ 3.8k FS 48 Use with care today
Data & Analysis ✓ NVIDIA · Official

TAO PyT Image Classification

Train, evaluate, optimize, and deploy PyTorch image classifiers with NVIDIA TAO.

★ 3.8k FS 48 Use with care today
Data & Analysis ✓ NVIDIA · Official

TAO MAE Vision Pretraining

Guides NVIDIA TAO through masked-autoencoder pretraining and fine-tuning for visual representations.

★ 3.8k FS 48 Use with care today
Data & Analysis ✓ NVIDIA · Official

TAO Deformable DETR Object Detection

Guides end-to-end training, evaluation, export, quantization, and deployment of lightweight Deformable DETR detectors in NVIDIA TAO.

★ 3.8k FS 46 Use with care today
Data & Analysis ✓ NVIDIA · Official

TAO OneFormer Universal Segmentation

Train and deploy one task-conditioned model for panoptic, instance, and semantic image segmentation.

★ 3.8k FS 53 Use with care today
Data & Analysis ✓ NVIDIA · Official

Mask Grounding DINO Training & Deployment

Train and deploy text-guided open-vocabulary instance segmentation models.

★ 3.8k FS 51 Use with care today
Data & Analysis ✓ NVIDIA · Official

TAO Mask2Former Segmentation

Guides NVIDIA TAO workflows for training and deploying Mask2Former segmentation models.

★ 3.8k FS 49 Use with care today
Data & Analysis ✓ NVIDIA · Official

TAO Visual ChangeNet Inspection

Train paired-image classifiers and change-segmentation models for PCB defect inspection.

★ 3.8k FS 48 Use with care today
Data & Analysis ✓ NVIDIA · Official

TAO DAFT Dataset Validator

Validate TAO DAFT dataset structure, schemas, and cross-references with tao-daft validate.

★ 3.8k FS 59 Recommended today
Data & Analysis ✓ NVIDIA · Official

TAO CLIP Fine-Tuning and Deployment

Train and deploy NVIDIA TAO CLIP for image-text retrieval, zero-shot classification, and embedding extraction.

★ 3.8k FS 55 Use with care today
Data & Analysis ✓ NVIDIA · Official

TAO FastFoundationStereo Real-Time Stereo Depth

Run TAO FastFoundationStereo workflows that produce low-latency disparity maps from stereo images.

★ 3.8k FS 55 Use with care today
Data & Analysis ✓ NVIDIA · Official

Cosmos-Embed Video Retrieval & Fine-Tuning

Use Cosmos-Embed1 to embed, retrieve, deduplicate, and fine-tune video-text data.

★ 3.8k FS 53 Use with care today
Data & Analysis ✓ NVIDIA · Official

TAO FoundationStereo Stereo Depth

Generate disparity maps from stereo images and manage FoundationStereo training, evaluation, and deployment for 3D reconstruction.

★ 3.8k FS 52 Use with care today
Data & Analysis ✓ NVIDIA · Official

TAO AutoML + DEFT Three-Phase Pipeline

Chains hyperparameter tuning, iterative data improvement, and final refinement for AOI and other DEFT applications.

★ 3.8k FS 51 Use with care today
Data & Analysis ✓ NVIDIA · Official

TAO Person Re-Identification

Train and deploy TAO ReID models for cross-camera person matching.

★ 3.8k FS 49 Use with care today
Data & Analysis ✓ NVIDIA · Official

TAO NVDINOv2 Visual Representation Training

Train reusable visual features with label-free teacher-student distillation.

★ 3.8k FS 47 Use with care today

Related skills