Skip to content

Add resiliency examples for ft launcher and straggler detection#2115

Merged
ananthsub merged 2 commits intoNVIDIA-NeMo:mainfrom
ananthsub:resiliency-examples
Feb 2, 2026
Merged

Add resiliency examples for ft launcher and straggler detection#2115
ananthsub merged 2 commits intoNVIDIA-NeMo:mainfrom
ananthsub:resiliency-examples

Conversation

@ananthsub
Copy link
Copy Markdown
Contributor

@ananthsub ananthsub commented Jan 29, 2026

What does this PR do ?

These serve as code examples to accompany the documentation here: https://docs.nvidia.com/nemo/megatron-bridge/latest/training/resiliency.html

The examples here highlight the straggler detection and fault tolerant job launching (directly using the ft_launcher) integrations through NVRx. These examples provide a starting point similar to what nemo2 offered here: https://github.com/NVIDIA-NeMo/NeMo/tree/main/examples/llm/resiliency

and what was tested against in nvrx here: https://github.com/NVIDIA/nvidia-resiliency-ext/tree/main/tests/ptl_resiliency/func/nemo20

Changelog

  • Add resiliency examples for fault tolerant job launching and straggler detection

GitHub Actions CI

See the CI sectionin the Contributing doc for how to trigger the CI. A Nvidia developer will need to approve and trigger the CI for external contributors.

Before your PR is "Ready for review"

Pre checks:

  • Make sure you read and followed Contributor guidelines
  • Did you write any new necessary tests?
  • Did you add or update any necessary documentation?
  • Does the PR affect components that are optional to install? (Ex: Numba, Pynini, Apex etc)
    • Reviewer: Does the PR have correct import guards for all optional libraries?

If you haven't finished some of the above items you can still open "Draft" PR.

Additional Information

  • Related to # (issue)

Summary by CodeRabbit

  • Documentation

    • Added resiliency features documentation including prerequisites and use case guides.
  • New Features

    • Added straggler detection example code and deployment scripts for distributed training.
    • Added fault-tolerance example code with automatic hang detection, job restart capabilities, and optional fault simulation capabilities.

✏️ Tip: You can customize this high-level summary in your review settings.

Signed-off-by: Ananth Subramaniam <ansubramania@nvidia.com>
@ananthsub ananthsub requested a review from maanug-nv January 29, 2026 00:58
@copy-pr-bot
Copy link
Copy Markdown

copy-pr-bot bot commented Jan 29, 2026

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

@ananthsub
Copy link
Copy Markdown
Contributor Author

/ok to test 8623ba6

@coderabbitai coderabbitai bot mentioned this pull request Feb 12, 2026
5 tasks
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants