NVIDIA Resiliency Extension

The NVIDIA Resiliency Extension (NVRx) integrates multiple resiliency-focused solutions for PyTorch-based workloads. Users can modularly integrate NVRx capabilities into their own infrastructure to maximize AI training productivity at scale. NVRx maximizes goodput by enabling system-wide health checks, quickly detecting faults at runtime and resuming training automatically. NVRx minimizes loss of work by enabling fast and frequent checkpointing.

For detailed documentation and usage information about each component, please refer to https://nvidia.github.io/nvidia-resiliency-ext/.

⚠️ NOTE: This project is still experimental and under active development. The code, features, and documentation are evolving rapidly. Please expect frequent updates and breaking changes. Contributions are welcome and we encourage you to watch for updates.

Core Components and Capabilities

• Fault Tolerance

  • Detection of hung ranks.
  • Restarting training in-job, without the need to reallocate SLURM nodes.

• In-Process Restarting

  • Detecting failures and enabling quick recovery.

• Async Checkpointing

  • Providing an efficient framework for asynchronous checkpointing.

• Local Checkpointing

  • Providing an efficient framework for local checkpointing.

• Straggler Detection

  • Monitoring GPU and CPU performance of ranks.
  • Identifying slower ranks that may impede overall training efficiency.

• Framework Integration

  • Facilitating seamless fault tolerance and straggler detection integration with PyTorch Lightning based workloads.
  • Providing integration with NVIDIA NeMo framework, a scalable and cloud-native generative AI framework built for researchers and developers working on Large Language Models, Multimodal, and Speech AI (e.g. Automatic Speech Recognition and Text-to-Speech).

Installation

From sources

• git clone https://github.com/NVIDIA/nvidia-resiliency-ext
• cd nvidia-resiliency-ext
• pip install .
• pip install .[attribution] if you also need log-analysis / attribution extras

From PyPI wheel

• pip install nvidia-resiliency-ext
• pip install 'nvidia-resiliency-ext[attribution]' for attribution extras

Platform Support