Skip to content
Navigation Menu
Sign in
Appearance settings
Platform
AI CODE CREATION
GitHub Copilot
Write better code with AI
GitHub Copilot app
Direct agents from issue to merge
MCP Registry
Integrate external tools
DEVELOPER WORKFLOWS
Actions
Automate any workflow
Codespaces
Instant dev environments
Issues
Plan and track work
Code Review
Manage code changes
Code Quality
Enforce quality at merge
APPLICATION SECURITY
GitHub Advanced Security
Find and fix vulnerabilities
Code security
Secure your code as you build
Secret protection
Stop leaks before they start
EXPLORE
Why GitHub
Documentation
Blog
Changelog
Marketplace
View all features
Solutions
BY COMPANY SIZE
Enterprises
Small and medium teams
Startups
Nonprofits
BY USE CASE
App Modernization
DevSecOps
DevOps
CI/CD
View all use cases
BY INDUSTRY
Healthcare
Financial services
Manufacturing
Government
View all industries
View all solutions
Resources
EXPLORE BY TOPIC
AI
Software Development
DevOps
Security
View all topics
EXPLORE BY TYPE
Customer stories
Events & webinars
Ebooks & reports
Business insights
GitHub Skills
SUPPORT & SERVICES
Documentation
Customer support
Community forum
Trust center
Partners
View all resources
Open Source
COMMUNITY
GitHub Sponsors
Fund open source developers
PROGRAMS
Security Lab
Maintainer Community
Accelerator
GitHub Stars
Archive Program
REPOSITORIES
Topics
Trending
Collections
Enterprise
ENTERPRISE SOLUTIONS
Enterprise platform
AI-powered developer platform
AVAILABLE ADD-ONS
GitHub Advanced Security
Enterprise-grade security features
Copilot for Business
Enterprise-grade AI features
Premium Support
Enterprise-grade 24/7 support
Pricing
Type
/
to search
Sign in
Sign up
Appearance settings
You signed in with another tab or window.
Reload
to refresh your session.
You signed out in another tab or window.
Reload
to refresh your session.
You switched accounts on another tab or window.
Reload
to refresh your session.
Dismiss alert
{{ message }}
Uh oh!
There was an error while loading.
Please reload this page
.
NVIDIA
/
Megatron-LM
Public
Notifications
You must be signed in to change notification settings
Fork
4.4k
Star
17.4k
Code
Issues
381
Pull requests
781
Discussions
Actions
Projects
Security and quality
0
Insights
Additional navigation options
Code
Issues
Pull requests
Discussions
Actions
Projects
Security and quality
Insights
Support HSDP deferred DP-outer gradient reduction
- #5743
#5743
Merged
wujingyue
merged 17 commits into
NVIDIA:main
NVIDIA/Megatron-LM:main
from
Achyuthan-S:fsdp/hsdp-grad-reduce
Achyuthan-S/Megatron-LM:fsdp/hsdp-grad-reduce
Copy head branch name to clipboard
Jul 20, 2026
Conversation
Commits
17
(17)
Checks
Files changed
Merged
Support HSDP deferred DP-outer gradient reduction
#5743
wujingyue
merged 17 commits into
NVIDIA:main
NVIDIA/Megatron-LM:main
from
Achyuthan-S:fsdp/hsdp-grad-reduce
Achyuthan-S/Megatron-LM:fsdp/hsdp-grad-reduce
Copy head branch name to clipboard
Commits
Commits on Jul 10, 2026
Support HSDP grad reduce in experimental mFSDP fully_shard
Show description for 6095687
Achyuthan-S
committed
6095687
View commit details
Copy full SHA for 6095687
Browse repository at this point
Restrict deferred-grad guard to one axis and release stale reduced-grad buffer
Show description for 7248685
Achyuthan-S
committed
7248685
View commit details
Copy full SHA for 7248685
Browse repository at this point
Commits on Jul 11, 2026
Remove over-defensive main_grad/main_weight placement guard
Show description for 3d271f3
Achyuthan-S
committed
3d271f3
View commit details
Copy full SHA for 3d271f3
Browse repository at this point
Commits on Jul 13, 2026
Make deferred DP-outer gradient reduction stateless
Show description for 4ed16af
Achyuthan-S
committed
4ed16af
View commit details
Copy full SHA for 4ed16af
Browse repository at this point
Commits on Jul 14, 2026
Rest main_grad DP-outer-Partial between microbatches for HSDP
Show description for dac14f2
Achyuthan-S
committed
dac14f2
View commit details
Copy full SHA for dac14f2
Browse repository at this point
Commits on Jul 15, 2026
Apply Claude fix for PR #5743
Show description for 434a4d9
svcnvidia-nemo-ci
committed
434a4d9
View commit details
Copy full SHA for 434a4d9
Browse repository at this point
Address review: store accumulation placement, redistribute finalize, harden get_dtensor reduce op
Show description for 6570526
Achyuthan-S
committed
6570526
View commit details
Copy full SHA for 6570526
Browse repository at this point
Commits on Jul 16, 2026
Address review: drop symm-mem guard, DRY grad install, tidy reset
Show description for ba659e8
Achyuthan-S
committed
ba659e8
View commit details
Copy full SHA for ba659e8
Browse repository at this point
Address review: Replicate->Partial redistribute, restore reduce-into-main_grad
Show description for bd293b3
Achyuthan-S
committed
bd293b3
View commit details
Copy full SHA for bd293b3
Browse repository at this point
Merge upstream/main into fsdp/hsdp-grad-reduce
Show description for 6a55d0e
Achyuthan-S
committed
6a55d0e
View commit details
Copy full SHA for 6a55d0e
Browse repository at this point
Address review: zero-copy replicate_to_partial, dedup, DP-outer event test
Show description for 325f338
Achyuthan-S
committed
325f338
View commit details
Copy full SHA for 325f338
Browse repository at this point
Exercise root-unit overlap in the HSDP event-count test
Show description for 8bb6e18
Achyuthan-S
committed
8bb6e18
View commit details
Copy full SHA for 8bb6e18
Browse repository at this point
Exercise root-unit overlap in the HSDP parity test too
Show description for d0f8185
Achyuthan-S
committed
d0f8185
View commit details
Copy full SHA for d0f8185
Browse repository at this point
Commits on Jul 17, 2026
Address review: inline Replicate->Partial relabel, share event helper, stricter assert
Show description for 8b18bde
Achyuthan-S
committed
8b18bde
View commit details
Copy full SHA for 8b18bde
Browse repository at this point
Count only NCCL kernel events in _nccl_events
Show description for 96764cd
Achyuthan-S
committed
96764cd
View commit details
Copy full SHA for 96764cd
Browse repository at this point
Apply Claude fix for PR #5743
Show description for 89b5296
svcnvidia-nemo-ci
committed
89b5296
View commit details
Copy full SHA for 89b5296
Browse repository at this point
Commits on Jul 18, 2026
Apply Claude fix for PR #5743
Show description for 66b592a
svcnvidia-nemo-ci
committed
66b592a
View commit details
Copy full SHA for 66b592a
Browse repository at this point
You can’t perform that action at this time.
