Update distill Megatron plugin#319

AAnoosheh merged 3 commits intomainNVIDIA/TensorRT-Model-Optimizer:mainfrom

aanoosheh/update-megatron-kdNVIDIA/TensorRT-Model-Optimizer:aanoosheh/update-megatron-kdCopy head branch name to clipboard