What problem does it solve?
This Skill removes the engineering friction of running large multi-seed or multi-phase ML experiment grids over SSH by automatically scheduling jobs, handling OOM retries, cleaning stale screen sessions, and preventing wave-transition race conditions.
Core Features & Use Cases
- SSH GPU batch orchestration: Schedules 10+ jobs across available remote GPUs using a manifest with explicit job states.
- OOM-aware retry & bounded failure: Detects CUDA out-of-memory in per-job logs, retries after a delay up to a configured max, then marks the job stuck for manual follow-up.
- Stale screen cleanup & idempotent resume: Detects cases where a screen remains but the Python process exited, cleans up safely, continuously persists
queue_state.json, and supports restarting the scheduler without losing progress.
- Wave transitions & dependency phases: Runs jobs in waves and enforces phase dependencies (e.g., teacher training must complete before student distillation).
Quick Start
Ask the AI to run experiment-queue for your multi-seed or multi-phase grid by providing a grid spec or manifest and the remote SSH GPU target (e.g., “Run experiment-queue for a 21-seed sweep across 12 configs with teacher→student phases, retry OOM up to 3 times, and write results to the specified remote run directory”).