Tao Run On Slurm

Bridges TAO training jobs to on-prem SLURM clusters over SSH. You give it credentials (SLURM_USER, SLURM_HOSTNAME, SSH key), point it at Lustre datasets, and it handles the sbatch submission, Pyxis container execution, and status polling. Built for DGX clusters with shared storage. The preflight checks are thorough, including a heredoc trick to install NGC credentials without leaking your key into shell history. Multi-node PyTorch jobs get the right WORLD_SIZE and rendezvous variables automatically. Default wall time is 4 hours, not 12, which matches the packaged partition list. If you're running TAO workloads on managed GPU infrastructure instead of local docker, this is the connector.

Install to Claude Code

npx -y skills add nvidia/skills --skill tao-run-on-slurm --agent claude-code