nemo-mbridge-perf-megatron-fsdp
CommunityEnable Megatron FSDP in Megatron-Bridge.
Software Engineering#distributed-training#nvidia#memory-efficiency#megatron-bridge#megatron-fsdp#checkpoint-format
Authorsayalinvidia
Version1.0.0
Installs0
System Documentation
What problem does it solve?
Megatron-Bridge users face memory constraints when training large models; enabling Megatron FSDP helps mitigate OOM errors and improves scalability.
Core Features & Use Cases
- Enables Megatron FSDP override in Megatron-Bridge for memory-efficient data parallelism.
- Provides code anchors, configuration guidance, pitfall notes, and verification steps to ensure correct setup.
- Use Case: diagnose memory regressions and optimize checkpoint formats for FSDP-based training.
Quick Start
Run the Megatron FSDP override in Megatron-Bridge and perform a small validation to confirm the ckpt_format is fsdp_dtensor and the mutual exclusion with Torch FSDP2.
Dependency Matrix
Required Modules
None requiredComponents
Standard package💻 Claude Code Installation
Recommended: Let Claude install automatically. Simply copy and paste the text below to Claude Code.
Please help me install this Skill: Name: nemo-mbridge-perf-megatron-fsdp Download link: https://github.com/sayalinvidia/sayali-skills-test/archive/main.zip#nemo-mbridge-perf-megatron-fsdp Please download this .zip file, extract it, and install it in the .claude/skills/ directory.
Agent Skills Search Helper
Install a tiny helper to your Agent, search and equip skill from 510,000+ vetted skills library on demand.