TAO Execution SDK 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: python -c "import tao_sdk" 2
/dev/null || { echo "Installing missing Python requirement: nvidia-tao-sdk[all]" python -m pip install "nvidia-tao-sdk[all]" } python -c "import tao_sdk" Show more Installs 586 Repository nvidia/skills GitHub Stars 1.9K First Seen Jun 8, 2026 Security Audits Gen Agent Trust Hub Pass Socket Pass Snyk Warn