Developer Tools

Amazon and NVIDIA Fixed the Glitch That Wastes Weeks of AI Training

When one AI chip hiccups, the whole job dies — now it recovers in seconds.

Deep Dive

NVIDIA's Resiliency Extension (NVRx) can be integrated into PyTorch Fully Sharded Data Parallel training on Amazon EKS to keep large distributed training jobs making progress when things break. At the scale described — jobs running for hours or days across dozens of nodes — interruptions are statistically inevitable, and a single GPU fault can cascade through NCCL timeouts, crash pods out of sync, and burn expensive GPU hours while making zero training progress. NVRx adds fault-tolerance primitives as ordinary pip-installable Python imports, with no custom kernels, no PyTorch fork, and no recompile: async checkpointing that overlaps I/O with training, in-process restart that recovers from faults in seconds without touching the container lifecycle, and ft_launcher for automatic worker respawn on hard crashes like SIGKILL and OOM kills. On the cluster sizes covered in the article, synchronous checkpointing alone consumed up to 40% of total wall time.

Key Points
  • One broken chip used to crash an entire multi-day AI training job — now the system heals itself in seconds.
  • Saving progress in the background instead of pausing could cut up to 40% of wasted computing time.
  • The tool is free and open source, tested on expensive H100 chips rented through Amazon's cloud.

Why It Matters

Faster, cheaper AI training means better AI tools reach you sooner — and less energy wasted on restarts.

📬 Get the top 10 AI stories daily