forked from NVIDIA/Megatron-LM
-
Notifications
You must be signed in to change notification settings - Fork 50
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Updated Megatron version #85
Draft
jlamypoirier
wants to merge
428
commits into
bigcode-project:nvidia_main
Choose a base branch
from
ServiceNow:compare_tensors_updated
base: nvidia_main
Could not load branches
Branch not found: {{ refName }}
Loading
Could not load tags
Nothing to show
Loading
Are you sure you want to change the base?
Some commits from the old base branch may be removed from the timeline,
and old review comments may become outdated.
Draft
Updated Megatron version #85
jlamypoirier
wants to merge
428
commits into
bigcode-project:nvidia_main
from
ServiceNow:compare_tensors_updated
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sliding window attention See merge request ADLR/megatron-lm!1025
Refactor DistributedOptimizer for MoE model support See merge request ADLR/megatron-lm!986
Signed-off-by: Selvaraj Anandaraj <[email protected]>
Signed-off-by: jiemingz <[email protected]>
Signed-off-by: jiemingz <[email protected]>
Signed-off-by: jiemingz <[email protected]>
Signed-off-by: Selvaraj Anandaraj <[email protected]>
add is_first_microbatch for TE See merge request ADLR/megatron-lm!1033
Need a switch at NeMo level to enable Atomic GEMM See merge request ADLR/megatron-lm!1017
Add distributed checkpoint support to non-TE based models See merge request ADLR/megatron-lm!1005
Signed-off-by: Selvaraj Anandaraj <[email protected]>
Signed-off-by: Selvaraj Anandaraj <[email protected]>
Support for activation offloading to CPU in M-LM See merge request ADLR/megatron-lm!1016
add rope and swiglu fusion See merge request ADLR/megatron-lm!946
Mcore CLIP ViT model See merge request ADLR/megatron-lm!1127
Bugfix: Make sure data_end_index is padded when creating new buckets See merge request ADLR/megatron-lm!1140
Unify resume and correctness functional tests See merge request ADLR/megatron-lm!1070
Mcore mock multimodal dataset See merge request ADLR/megatron-lm!1147
…ommunication Compute norm once per batch (instead of once per microbatch) and once per bucket (instead of once per param)
Fix NaN checking in grads: should be performed before data-parallel all-reduce See merge request ADLR/megatron-lm!989
…ry ffn_hidden_size
Move to Draco OCI See merge request ADLR/megatron-lm!1137
Print number of transformer and embedding parameters separately See merge request ADLR/megatron-lm!1159
Mcore LLaVA model See merge request ADLR/megatron-lm!1151
[OMNIML-614] AMMO ptq + TensorRT-LLM export examples for megatron-lm See merge request ADLR/megatron-lm!1013
Make throughput and memory footprint formulae compatible with arbitrary ffn_hidden_size See merge request ADLR/megatron-lm!1169
Experimental Yaml configs See merge request ADLR/megatron-lm!1134
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
No description provided.