Agent skill · NVIDIA
tao-run-platform
TAO Execution SDK for submitting and monitoring GPU training jobs on supported platforms (Brev, SLURM,
What it needs
About 8k tokens when loaded.
What this skill does
TAO Execution SDK Standalone install? If this session was not initialized by the TAO skill bank plugin, run the tao-setup skill first (host preflight, credentials, cross-skill discovery). The SDK is the optional Python layer for users who need job handles, S3 I/O wrapping, or platform-specific features (SLURM/Lustre queues, Kubernetes Jobs, local Docker debugging, Brev instance reuse). Most TAO skills run with just docker run and don't need it. Reach for the SDK when: You want a Job handle to poll status and stream logs over time. You need S3-aware input download / output upload baked into the entrypoint. You're chaining multiple jobs and want persisted state. Preflight Install nvidia-tao-sdk[all] before using this platform — the [all] extra pulls in every platform-specific dependency (Brev, S3 utilities, etc.). If it is missing, install it by default in the active Python environment and rerun the import check: The package index is environment-specific — the runner/container is expected to have a working pip configuration (e.g. ~/.pip/pip.conf, PIPINDEXURL, PIPEXTRAINDEXURL, or proxy). If the install fails for index/network reasons, that's a runner setup issue; this skill stays agnostic to the registry. Missing pip requirements are installed automatically by default and reported in the run log. Non-pip/system prerequisites still require a normal preflight failure and user-visible remediation. Setup Credentials come from environment variables — read from the session environment (export them in your shell before launching). The SDK validates credentials lazily on first use and raises CredentialError with a clear message if a required env var is missing. Required env vars: Platform Required Optional --- --- --- Brev — (manual brev login works) BREVAPITOKEN S3 I/O (any platform) S3BUCKETNAME, ACCESSKEY, SECRETKEY S3ENDPOINTURL, CLOUDREGION Container env NGCKEY HFTOKEN The agent never reads credential values — it only checks presence with [ -n "$VARNAME" ]. …
How to use it
Reference it in AdaL, Claude Code, Cursor or any coding agent — nothing to install:
@skills NVIDIA/tao-run-platform