NVIDIA/nvidia-resiliency-ext

NVIDIA/nvidia-resiliency-ext

Description: NVIDIA Resiliency Extension is a python package for framework developers and users to implement fault-tolerant features. It improves the effective training time by minimizing the downtime due to failures and interruptions.

Language: Python

License: NOASSERTION

Stars: 296

Forks: 53

Open issues: 24

Created: 2024-10-04T18:31:44Z

Pushed: 2026-06-10T17:02:44Z

Default branch: main

Fork: no

Archived: no

README:

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](https://nvidia.github.io/nvidia-resiliency-ext/fault_tolerance/index.html)
• Detection of hung ranks.
• Restarting training in-job, without the need to reallocate SLURM nodes.

• [In-Process Restarting](https://nvidia.github.io/nvidia-resiliency-ext/inprocess/index.html)
• Detecting failures and enabling quick recovery.

• [Async Checkpointing](https://nvidia.github.io/nvidia-resiliency-ext/checkpointing/async/index.html)
• Providing an efficient framework for asynchronous checkpointing.

• [Local Checkpointing](https://nvidia.github.io/nvidia-resiliency-ext/checkpointing/local/index.html)
• Providing an efficient framework for local checkpointing.

• [Straggler Detection](https://nvidia.github.io/nvidia-resiliency-ext/straggler_det/index.html)
• 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

| Category | Supported Versions / Requirements | |----------------------|----------------------------------------------------------------------------| | Architecture | x86_64, arm64 | | Operating System | Ubuntu 22.04, 24.04 | | Python Version | >= 3.10, = 2.5.1, >= 2.8.0 (Fault Attribution) | | CUDA & CUDA Toolkit | >= 12.8 | | NVML Driver | >= 535 (570 required for GPU health check) | | NCCL Version | = 2.28.9 (avoid NCCL 2.28.3–2.28.8 due to inprocess issue) | | TE Version | >= 2.5 |