Cherry pick NIM supporting changes for nemo.export for NeMo 2.0 (part II) (11669) into r2.1.0

Cherry pick NIM supporting changes for nemo.export for NeMo 2.0 (part II) (11669) into r2.1.0 #3106