Perf Env Plugin Cherry-pick
Cherry picks https://github.com/NVIDIA/NeMo/commit/b77c74302ab81ba4e9ee21ccf800f852b6fcad96 and https://github.com/NVIDIA/NeMo/commit/d467cd88d20fc716c8de8732bd1a0891527e8a26