What problem does it solve?
This Skill solves the problem of running reliable ML training and inference workloads on reserved, on-demand GPU instances with straightforward SSH access and persistent storage.
Core Features & Use Cases
- GPU cloud instance launch and SSH access: Choose GPU type and region, start an instance, and connect via SSH for hands-on experimentation or production jobs.
- Persistent filesystems for data and checkpoints: Keep datasets, checkpoints, and outputs across restarts by mounting Lambda filesystems at a stable path.
- Distributed and multi-node training support: Prepare for single-node multi-GPU and multi-node (Slurm/torchrun) training using pre-installed high-performance ML stack components.
Use case: You need an H100 cluster-style environment to fine-tune a large model and reliably resume from checkpoints stored on a persistent filesystem.
Quick Start
Tell the agent to set up an H100 instance in your chosen region, mount a filesystem for persistent checkpoints, then verify CUDA and PyTorch are working by running a GPU check and a CUDA availability check.