Merge branch 'dnarayanan/fix_import' into 'main'

Add missing import to megatron/training/initialize.py

See merge request ADLR/megatron-lm!2226

Author: deepakn94

megatron/training/initialize.py

@@ -5,6 +5,7 @@
 import random
 import os
 import time
+import warnings
 
 import numpy as np
 import torch