[Megatron-LM] Add unit tests to test Mamba and Zebra-Llama pretraining #595
Draft
clairesonglee wants to merge 12 commits into
Draft
[Megatron-LM] Add unit tests to test Mamba and Zebra-Llama pretraining #595clairesonglee wants to merge 12 commits into
clairesonglee wants to merge 12 commits into