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
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.5k
Code
Issues
395
Pull requests
812
Discussions
Actions
Projects
Security and quality
0
Insights
Additional navigation options
Code
Issues
Pull requests
Discussions
Actions
Projects
Security and quality
Insights
MimoOptimizer: fix distributed checkpoint save and load for non-colocated MIMO
- #4801
#4801
Merged
ericharper
merged 6 commits into
NVIDIA:main
NVIDIA/Megatron-LM:main
from
kamran-nvidia:kamran/mimo_optimizer_fixes
kamran-nvidia/Megatron-LM:kamran/mimo_optimizer_fixes
Copy head branch name to clipboard
May 21, 2026
Conversation
Commits
6
(6)
Checks
Files changed
Merged
MimoOptimizer: fix distributed checkpoint save and load for non-colocated MIMO
#4801
ericharper
merged 6 commits into
NVIDIA:main
NVIDIA/Megatron-LM:main
from
kamran-nvidia:kamran/mimo_optimizer_fixes
kamran-nvidia/Megatron-LM:kamran/mimo_optimizer_fixes
Copy head branch name to clipboard
Commits
Commits on May 14, 2026
Enhance MimoOptimizer and tests for distributed checkpointing support
Show description for 71e1265
kamran-nvidia
committed
71e1265
View commit details
Copy full SHA for 71e1265
Browse repository at this point
Commits on May 15, 2026
Refactor MimoOptimizer to enhance state management for param_state_sharding_type in distributed checkpointing
Show description for 7edaa8c
kamran-nvidia
committed
7edaa8c
View commit details
Copy full SHA for 7edaa8c
Browse repository at this point
Commits on May 19, 2026
revert PR #4791 fixes
Show description for 2f7158a
kamran-nvidia
committed
2f7158a
View commit details
Copy full SHA for 2f7158a
Browse repository at this point
revert changes to the tests
Show description for 927d083
kamran-nvidia
committed
927d083
View commit details
Copy full SHA for 927d083
Browse repository at this point
Merge branch 'main' into kamran/mimo_optimizer_fixes
kamran-nvidia
authored
bec6bbb
View commit details
Copy full SHA for bec6bbb
Browse repository at this point
Commits on May 20, 2026
Merge branch 'main' into kamran/mimo_optimizer_fixes
kamran-nvidia
authored
42453c0
View commit details
Copy full SHA for 42453c0
Browse repository at this point
You can’t perform that action at this time.
