Dev & Engineering ✓ NVIDIA · Official

TAO Platform Execution SDK

Submit, monitor, and manage NVIDIA TAO GPU training jobs across supported platforms.

55/ 100
Use with care

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

See how it was scored ↓
Works as-is in
Codex · Claude Code
Stars
★ 3.8k
Last updated
today
License
Apache-2.0
taogpu-trainingjob-monitorings3-io
+2kubernetesslurm

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

This skill guides use of the NVIDIA TAO Execution SDK on Brev, SLURM, local Docker, and Kubernetes. It enables job handles for status polling, log streaming, cancellation, failure analysis, and result retrieval. The SDK also wraps S3 input downloads and output uploads and supports multi-node and platform-specific execution features. It is best suited to TAO workflows that need durable tracking or platform integration beyond a simple docker run.

Checks for and installs nvidia-tao-sdk[all]; verifies the presence of environment variables without reading credential values; obtains the credential requirements for the selected platform; builds container commands with build_entrypoint; creates, monitors, cancels, and inspects jobs through the SDK; downloads declared inputs from S3, HF Hub, or NGC and uploads declared outputs; resolves result destinations using mounted storage or S3; and supports Brev instance listing and deletion.

Good fit
  • A TAO user needs to launch training on a Brev GPU instance and monitor it over time.
  • A team needs to submit multi-node distributed training through SLURM and Lustre queues.
  • An engineer wants S3 inputs and outputs handled automatically around a TAO container run.
  • A developer needs to debug TAO jobs through Kubernetes Jobs or local Docker.
  • A user wants to chain multiple TAO jobs while preserving job IDs and run state across context breaks.

How do you install this skill?

Before you use it
  • Before execution, manually confirm the platform, image, credential scope, data URIs, output persistence, GPU/cloud cost, and whether automatic dependency installation is permitted.
  • Pin and review nvidia-tao-sdk[all] and its platform dependencies, and verify reachability of required package indexes, container registries, S3, HF Hub, or NGC.
  • Do not pass agent-host local paths directly to SLURM or Kubernetes; after launch, verify that outputs are persisted rather than relying on container-ephemeral storage.
  • The benchmark covers only one positive description task and does not establish real TAO-job correctness, reliability, or security.
Before you start
Your agent needs
  • Shell / CLI
  • Network access
  • Local filesystem
Install first
  • Python 3.10+
  • nvidia-tao-sdk[all]
  • pip

Install the skill with the NVIDIA skills CLI:

npx skills add nvidia/skills --skill tao-run-platform --yes

It becomes available when the agent next loads skills and encounters a relevant task. Before using it, install the SDK dependency:

python -m pip install "nvidia-tao-sdk[all]"

The README does not state a separate price for this skill.

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-run-platform" ~/.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:

  • use the TAO SDK
  • TAO platform run,

Ask the agent to “use the TAO SDK” or perform a “TAO platform run,” and describe the TAO workflow. Before launch, select Brev, SLURM, local Docker, or Kubernetes; choose whether long-running monitoring should remain enabled and set the update interval, which defaults to five minutes. Confirm the resolved container image or provide image=<override>, then supply only the environment variables required by the selected platform and data URIs. The agent creates the job and continues polling until completion, failure, cancellation, or an explicit request to detach.

What are this skill's strengths and limitations?

Pros
  • One execution pattern covers Brev, SLURM, local Docker, and Kubernetes.
  • Provides job handles, status polling, logs, cancellation, failure analysis, and result lookup.
  • Includes S3-aware input and output handling.
  • Persists job IDs and state paths so background jobs can be reattached later.
  • Scopes credential requests by platform and does not read credential values.
Limitations
  • Requires Python 3.10+, nvidia-tao-sdk[all], and a working pip configuration.
  • Platform, S3, and container execution may require additional environment variables or login credentials.
  • Without a persistent mount or S3 bucket, outputs remain in an ephemeral container directory.
  • Long-running monitoring requires the chat or runtime to remain available, otherwise status must be refreshed manually.
  • The supplied source provides no skill-specific test suite or independent evidence for every platform.

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 Platform Execution SDK this page ✓ NVIDIA · Official 55 · Use with care ★ 3.8k today Apache-2.0
NeMo-RL Auto Research ✓ NVIDIA · Official 51 · Use with care ★ 3.8k today Apache-2.0
I4H Policy Fine-Tuning ✓ NVIDIA · Official 48 · Use with care ★ 3.8k today Apache-2.0
NeMo MBridge Training Guide ✓ NVIDIA · Official 44 · Not recommended ★ 3.8k today Apache-2.0
TAO Sparse4D Training Skill ✓ NVIDIA · Official 54 · Use with care ★ 3.8k today Apache-2.0

Compared with running TAO directly through docker run, this skill is aimed at workflows needing job handles, persistent monitoring, S3 I/O wrapping, scheduler integration, or multi-job orchestration. Most simple TAO runs do not need the SDK.

How did FollowSkills review this skill?

FollowSkills review · FSRS-2.0
Use with care
55/ 100 5-point scale 2.8 / 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.
1Trust16 / 25 · 3.2/5

The documentation limits credential handling to environment-variable presence checks, scopes credentials by platform, and explains S3/HF/NGC data flow, output locations, cancellation, and instance cleanup. However, it auto-installs an unpinned nvidia-tao-sdk[all] by default, creating network and supply-chain side effects, and lacks an explicit final confirmation immediately before launching a GPU job, so points are deducted.

2Reliability8 / 20 · 2.0/5

Core APIs, platform differences, polling, error patterns, and failure feedback are documented reasonably well. However, dependency versions are not pinned, key helpers, skill_info.yaml, scripts, and SDK implementations are outside the selected skill evidence, and no key-path test suite is included; the score is therefore conservative under static-review limits.

3Adaptability11 / 15 · 3.7/5

The audience, trigger phrases, platform selection, monitoring preferences, scope boundaries, and local-Docker multi-node limitation are fairly clear. Chinese-language guidance and mainland-China reachability for package indexes, registries, S3, HF, and NGC are not addressed, limiting environment-fit evidence.

4Convention10 / 15 · 3.3/5

The documentation is reasonably layered and includes preflight guidance, references, error patterns, scope boundaries, examples, Apache-2.0 licensing, version 0.1.0, and NVIDIA ownership. It lacks the recommended Instructions and Examples sections, a changelog or explicit update path, and exposes some dependency assumptions, so points are deducted.

5Effectiveness6 / 15 · 2.0/5

The skill could directly support complex TAO workflows involving submission, monitoring, logs, S3 I/O, multi-node execution, and multiple platforms. Static files do not verify real job outputs; the benchmark contains only one plan-description task, and platform prerequisites plus automatic installation can be costly, so the score remains below the static ceiling.

6Verifiability4 / 10 · 2.0/5

Committed evals.json, BENCHMARK.md, error mappings, and concrete code examples provide some auditability. However, there is only one evaluation task, no negative task or committed key-path tests, and the benchmark report is not independently reproducible from the supplied files, so coverage is thin.

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][12][13][14]

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

Does the skill read my cloud credential values?
No. It checks only whether required environment variables are present. Missing variables cause the SDK to raise CredentialError.
Which execution platforms are supported?
The SKILL.md names Brev, SLURM, local Docker, and Kubernetes.
What happens when a job fails?
The agent can retrieve recent logs and call get_failure_analysis() for an error class, remediation suggestion, and node-level failure events.
Can monitoring continue during a long queue wait?
Yes. With monitoring enabled, the workflow continues through long PENDING periods until the job reaches a terminal state or the user asks to stop.

More skills from this repository

All from NVIDIA/skills

Automation & Ops ✓ NVIDIA · Official

TAO Jobs on Kubernetes

Submit and monitor NVIDIA TAO GPU training jobs through Kubernetes.

★ 3.8k FS 53 Use with care today
Automation & Ops ✓ NVIDIA · Official

TAO Workflow Launch Gate

Preflight TAO jobs across platforms, credentials, data, containers, and monitoring.

★ 3.8k FS 60 Recommended today
Automation & Ops ✓ NVIDIA · Official

Megatron-LM on SLURM

Launch, monitor, and troubleshoot multi-node Megatron-LM training on SLURM GPU clusters.

★ 3.8k FS 56 Use with care today
Automation & Ops ✓ NVIDIA · Official

TAO on SLURM

Run TAO training, evaluation, and inference on a remote GPU cluster through SSH, SLURM, and Pyxis/Enroot.

★ 3.8k FS 55 Use with care today
Automation & Ops ✓ NVIDIA · Official

NeMo-RL Kubernetes Launcher

Launch, monitor, iterate on, and debug NeMo-RL training runs on Kubernetes.

★ 3.8k FS 54 Use with care today
Automation & Ops ✓ NVIDIA · Official

NeMo MBridge Multi-Node Slurm

Convert single-node MBridge scripts into multi-node Slurm jobs and diagnose launch, NCCL, and MoE memory failures.

★ 3.8k FS 44 Blocked today
Automation & Ops ✓ NVIDIA · Official

TAO NVIDIA GPU Host Setup

Checks and standardizes NVIDIA drivers, CUDA, and container runtime prerequisites for TAO GPU hosts.

★ 3.8k FS 55 Use with care today
Automation & Ops ✓ NVIDIA · Official

NeMo AutoModel Launcher Configuration

Configure NeMo AutoModel launches across local, Slurm, and cloud runs.

★ 3.8k FS 54 Use with care today
Automation & Ops ✓ NVIDIA · Official

Dynamo Deployment Troubleshooter

Diagnose unhealthy Dynamo deployments on Kubernetes from read-only evidence.

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

NeMo-RL Auto Research

Turns NeMo-RL research goals into reproducible, iterative experiments.

★ 3.8k FS 51 Use with care today
Automation & Ops ✓ NVIDIA · Official

Dynamo Recipe Runner

Select, validate, minimally patch, and deploy existing NVIDIA Dynamo inference recipes on Kubernetes.

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

I4H Policy Fine-Tuning

Fine-tune GR00T or openpi PI0 policies on recorded LeRobot demonstrations.

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

PAIDF AnomalyGen

Fine-tune, generate, evaluate, and refine synthetic anomaly images.

★ 3.8k FS 47 Use with care today
Automation & Ops ✓ NVIDIA · Official

TAO Inference Microservice Deployment

Start, query, and stop TAO inference microservices across supported compute platforms.

★ 3.8k FS 45 Not recommended today
Dev & Engineering ✓ NVIDIA · Official

NeMo MBridge Training Guide

Run Megatron-LM and Megatron Bridge training and verify whether their loss curves match.

★ 3.8k FS 44 Not recommended today
Automation & Ops ✓ NVIDIA · Official

DOCA Management Service Operations

Operate and troubleshoot NVIDIA DMS for centralized BlueField and ConnectX management.

★ 3.8k FS 55 Use with care today
Automation & Ops ✓ NVIDIA · Official

Dynamo Router Starter

Configure Dynamo routing modes and verify frontend endpoints quickly.

★ 3.8k FS 53 Use with care today
Automation & Ops ✓ NVIDIA · Official

Dynamo Interconnect Readiness Check

Verify RDMA, NVLink, and NIXL transport readiness before trusting performance results from disaggregated Dynamo serving.

★ 3.8k FS 52 Use with care today
Automation & Ops ✓ NVIDIA · Official

Physical AI Infrastructure Resilience

Set up, validate, scale, and recover Physical AI infrastructure for synthetic data generation workflows.

★ 3.8k FS 47 Use with care today
Automation & Ops ✓ NVIDIA · Official

NVIDIA RAG Blueprint Operations

Deploy, configure, troubleshoot, and manage NVIDIA RAG Blueprint environments.

★ 3.8k FS 39 Not recommended today

Related skills