search

aws-neuron / neuronx-nemo-megatron

32 stars 11 forks source link

Fix index mismatch for PP 1 on MPIJob and add reference hf ckpt #32

Closed amithrm closed 3 months ago

amithrm commented 3 months ago

Issue #, if available:

Description of changes:

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.