Add PP support in NeVA along with few bug fixes
<!-- pylint-output-strict-mode-true -->

beep boop 🤖: 🚨 The following files must be fixed before merge!

---

Your code was analyzed with PyLint. The following annotations have been identified:

```

------------------------------------
Your code has been rated at 10.00/10
```

---

Thank you for improving NeMo's documentation! <!-- pylint-output-strict-mode-false -->

beep boop 🤖: 🙏 The following files have warnings. In case you are familiar with these, please try helping us to improve the code base.

---

Your code was analyzed with PyLint. The following annotations have been identified:

```
************* Module nemo.collections.vlm.neva.data.llava_next_energon
nemo/collections/vlm/neva/data/llava_next_energon.py:28:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/vlm/neva/data/llava_next_energon.py:33:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/vlm/neva/data/llava_next_energon.py:37:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/vlm/neva/data/llava_next_energon.py:104:0: C0115: Missing class docstring (missing-class-docstring)
************* Module nemo.collections.llm.fn.activation
nemo/collections/llm/fn/activation.py:25:0: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/fn/activation.py:30:0: C0116: Missing function or method docstring (missing-function-docstring)
************* Module nemo.collections.llm.gpt.model.base
nemo/collections/llm/gpt/model/base.py:226:0: C0301: Line too long (122/119) (line-too-long)
nemo/collections/llm/gpt/model/base.py:51:0: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/base.py:94:0: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/base.py:115:0: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/base.py:123:0: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/base.py:131:0: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/base.py:139:0: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/base.py:149:0: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/base.py:159:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/base.py:182:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/base.py:252:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/base.py:264:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/base.py:276:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/base.py:288:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/base.py:300:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/base.py:312:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/base.py:326:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/base.py:344:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/base.py:348:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/base.py:371:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/base.py:374:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/base.py:377:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/base.py:381:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/base.py:386:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/base.py:407:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/base.py:414:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/base.py:421:0: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/base.py:449:0: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/base.py:43:4: W0611: Unused import fused_weight_gradient_mlp_cuda (unused-import)
************* Module nemo.collections.llm.gpt.model.llama
nemo/collections/llm/gpt/model/llama.py:483:0: C0301: Line too long (158/119) (line-too-long)
nemo/collections/llm/gpt/model/llama.py:42:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/llama.py:62:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/llama.py:71:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/llama.py:80:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/llama.py:89:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/llama.py:111:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/llama.py:132:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/llama.py:142:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/llama.py:154:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/llama.py:164:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/llama.py:175:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/llama.py:186:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/llama.py:192:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/llama.py:198:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/llama.py:209:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/llama.py:213:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/llama.py:225:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/llama.py:245:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/llama.py:259:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/llama.py:265:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/llama.py:302:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/llama.py:321:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/llama.py:338:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/llama.py:342:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/llama.py:475:0: C0116: Missing function or method docstring (missing-function-docstring)
************* Module nemo.collections.llm.gpt.model.ssm
nemo/collections/llm/gpt/model/ssm.py:190:0: C0301: Line too long (129/119) (line-too-long)
nemo/collections/llm/gpt/model/ssm.py:193:0: C0301: Line too long (131/119) (line-too-long)
nemo/collections/llm/gpt/model/ssm.py:194:0: C0301: Line too long (151/119) (line-too-long)
nemo/collections/llm/gpt/model/ssm.py:195:0: C0301: Line too long (129/119) (line-too-long)
nemo/collections/llm/gpt/model/ssm.py:39:0: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/ssm.py:51:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/ssm.py:89:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/ssm.py:110:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/ssm.py:127:8: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/ssm.py:131:12: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/ssm.py:134:12: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/ssm.py:155:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/ssm.py:203:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/ssm.py:216:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/llm/gpt/model/ssm.py:221:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/ssm.py:235:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/ssm.py:249:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/ssm.py:263:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/ssm.py:277:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/ssm.py:291:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/llm/gpt/model/ssm.py:305:0: C0115: Missing class docstring (missing-class-docstring)
************* Module nemo.collections.vlm.mllama.data.mock
nemo/collections/vlm/mllama/data/mock.py:27:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/vlm/mllama/data/mock.py:68:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/mllama/data/mock.py:79:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/mllama/data/mock.py:84:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/mllama/data/mock.py:89:4: C0116: Missing function or method docstring (missing-function-docstring)
************* Module nemo.collections.vlm.mllama.model.base
nemo/collections/vlm/mllama/model/base.py:50:0: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/mllama/model/base.py:99:0: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/mllama/model/base.py:116:0: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/mllama/model/base.py:121:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/vlm/mllama/model/base.py:152:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/mllama/model/base.py:155:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/mllama/model/base.py:162:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/vlm/mllama/model/base.py:227:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/vlm/mllama/model/base.py:248:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/mllama/model/base.py:276:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/vlm/mllama/model/base.py:305:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/mllama/model/base.py:315:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/mllama/model/base.py:319:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/vlm/mllama/model/base.py:361:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/mllama/model/base.py:364:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/mllama/model/base.py:398:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/mllama/model/base.py:491:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/vlm/mllama/model/base.py:508:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/mllama/model/base.py:512:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/mllama/model/base.py:541:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/mllama/model/base.py:544:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/mllama/model/base.py:547:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/mllama/model/base.py:551:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/mllama/model/base.py:557:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/mllama/model/base.py:564:4: C0116: Missing function or method docstring (missing-function-docstring)
************* Module nemo.collections.vlm.neva.data.conversation
nemo/collections/vlm/neva/data/conversation.py:93:0: C0301: Line too long (255/119) (line-too-long)
nemo/collections/vlm/neva/data/conversation.py:135:0: C0301: Line too long (198/119) (line-too-long)
nemo/collections/vlm/neva/data/conversation.py:358:0: C0301: Line too long (176/119) (line-too-long)
nemo/collections/vlm/neva/data/conversation.py:426:0: C0301: Line too long (314/119) (line-too-long)
nemo/collections/vlm/neva/data/conversation.py:428:0: C0301: Line too long (210/119) (line-too-long)
nemo/collections/vlm/neva/data/conversation.py:571:0: C0301: Line too long (152/119) (line-too-long)
nemo/collections/vlm/neva/data/conversation.py:595:0: C0301: Line too long (152/119) (line-too-long)
nemo/collections/vlm/neva/data/conversation.py:620:0: C0301: Line too long (152/119) (line-too-long)
nemo/collections/vlm/neva/data/conversation.py:63:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/data/conversation.py:67:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/data/conversation.py:78:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/data/conversation.py:237:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/data/conversation.py:240:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/data/conversation.py:286:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/data/conversation.py:300:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/data/conversation.py:324:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/data/conversation.py:336:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/data/conversation.py:18:0: W0611: Unused defaultdict imported from collections (unused-import)
************* Module nemo.collections.vlm.neva.data.lazy
nemo/collections/vlm/neva/data/lazy.py:564:0: C0301: Line too long (128/119) (line-too-long)
nemo/collections/vlm/neva/data/lazy.py:67:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/data/lazy.py:72:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/data/lazy.py:115:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/data/lazy.py:120:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/data/lazy.py:134:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/data/lazy.py:161:0: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/data/lazy.py:236:0: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/data/lazy.py:246:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/vlm/neva/data/lazy.py:418:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/data/lazy.py:493:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/vlm/neva/data/lazy.py:557:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/data/lazy.py:568:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/data/lazy.py:571:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/data/lazy.py:574:4: C0116: Missing function or method docstring (missing-function-docstring)
************* Module nemo.collections.vlm.neva.data.mock
nemo/collections/vlm/neva/data/mock.py:29:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/vlm/neva/data/mock.py:72:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/data/mock.py:83:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/data/mock.py:88:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/data/mock.py:93:4: C0116: Missing function or method docstring (missing-function-docstring)
************* Module nemo.collections.vlm.neva.model.api
nemo/collections/vlm/neva/model/api.py:20:0: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/model/api.py:24:0: C0116: Missing function or method docstring (missing-function-docstring)
************* Module nemo.collections.vlm.neva.model.base
nemo/collections/vlm/neva/model/base.py:470:0: C0301: Line too long (139/119) (line-too-long)
nemo/collections/vlm/neva/model/base.py:473:0: C0301: Line too long (128/119) (line-too-long)
nemo/collections/vlm/neva/model/base.py:481:0: C0301: Line too long (125/119) (line-too-long)
nemo/collections/vlm/neva/model/base.py:490:0: C0301: Line too long (122/119) (line-too-long)
nemo/collections/vlm/neva/model/base.py:93:0: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/model/base.py:134:0: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/model/base.py:151:0: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/model/base.py:172:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/model/base.py:227:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/model/base.py:242:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/vlm/neva/model/base.py:261:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/model/base.py:282:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/vlm/neva/model/base.py:313:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/model/base.py:351:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/model/base.py:364:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/vlm/neva/model/base.py:619:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/vlm/neva/model/base.py:636:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/model/base.py:640:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/model/base.py:664:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/model/base.py:667:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/model/base.py:670:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/model/base.py:674:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/model/base.py:680:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/model/base.py:687:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/model/base.py:27:0: W0611: Unused TEDotProductAttention imported from megatron.core.extensions.transformer_engine (unused-import)
************* Module nemo.collections.vlm.neva.model.llava
nemo/collections/vlm/neva/model/llava.py:108:0: C0301: Line too long (137/119) (line-too-long)
nemo/collections/vlm/neva/model/llava.py:109:0: C0301: Line too long (122/119) (line-too-long)
nemo/collections/vlm/neva/model/llava.py:110:0: C0301: Line too long (146/119) (line-too-long)
nemo/collections/vlm/neva/model/llava.py:111:0: C0301: Line too long (144/119) (line-too-long)
nemo/collections/vlm/neva/model/llava.py:146:0: C0301: Line too long (128/119) (line-too-long)
nemo/collections/vlm/neva/model/llava.py:147:0: C0301: Line too long (161/119) (line-too-long)
nemo/collections/vlm/neva/model/llava.py:148:0: C0301: Line too long (157/119) (line-too-long)
nemo/collections/vlm/neva/model/llava.py:149:0: C0301: Line too long (150/119) (line-too-long)
nemo/collections/vlm/neva/model/llava.py:150:0: C0301: Line too long (146/119) (line-too-long)
nemo/collections/vlm/neva/model/llava.py:151:0: C0301: Line too long (158/119) (line-too-long)
nemo/collections/vlm/neva/model/llava.py:152:0: C0301: Line too long (154/119) (line-too-long)
nemo/collections/vlm/neva/model/llava.py:153:0: C0301: Line too long (135/119) (line-too-long)
nemo/collections/vlm/neva/model/llava.py:154:0: C0301: Line too long (131/119) (line-too-long)
nemo/collections/vlm/neva/model/llava.py:155:0: C0301: Line too long (135/119) (line-too-long)
nemo/collections/vlm/neva/model/llava.py:156:0: C0301: Line too long (131/119) (line-too-long)
nemo/collections/vlm/neva/model/llava.py:41:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/vlm/neva/model/llava.py:46:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/vlm/neva/model/llava.py:59:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/vlm/neva/model/llava.py:71:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/vlm/neva/model/llava.py:83:0: C0115: Missing class docstring (missing-class-docstring)
nemo/collections/vlm/neva/model/llava.py:105:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/model/llava.py:177:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/model/llava.py:183:4: C0116: Missing function or method docstring (missing-function-docstring)
nemo/collections/vlm/neva/model/llava.py:223:0: C0116: Missing function or method docstring (missing-function-docstring)
************* Module nemo.lightning.io.state
nemo/lightning/io/state.py:31:0: C0115: Missing class docstring (missing-class-docstring)
nemo/lightning/io/state.py:334:4: C0116: Missing function or method docstring (missing-function-docstring)

-----------------------------------
Your code has been rated at 9.17/10
```

---

Thank you for improving NeMo's documentation! [🤖]: Hi @yaoyu-33 👋,

We wanted to let you know that a [CICD pipeline](https://github.com/NVIDIA/NeMo/actions/runs/11943673585) for this PR just finished successfully

So it might be time to merge this PR or get some approvals

I'm just a bot so I'll leave it you what to do next.

//cc @pablo-garay @ko3n1g
# What does this PR do ?

Add support for enabling PP in neva training. 2 Options are supplied: (1) vit share 1st pp stage with llm, encoder_pp=0.  (2) vit takes entire 1st pp stage with llm, encoder_pp=1.

**Collection**: [multimodal]

# Changelog 
- Add specific line by line info of high level changes in this PR.

# Usage
* You can potentially add a usage example below

```python
# Add a code snippet demonstrating how to use this 
```

# GitHub Actions CI

The Jenkins CI system has been replaced by GitHub Actions self-hosted runners.

The GitHub Actions CI will run automatically when the "Run CICD" label is added to the PR.
To re-run CI remove and add the label again.
To run CI on an untrusted fork, a NeMo user with write access must first click "Approve and run".

# Before your PR is "Ready for review"
**Pre checks**:
- [ ] Make sure you read and followed [Contributor guidelines](https://github.com/NVIDIA/NeMo/blob/main/CONTRIBUTING.md)
- [ ] Did you write any new necessary tests?
- [ ] Did you add or update any necessary documentation?
- [ ] Does the PR affect components that are optional to install? (Ex: Numba, Pynini, Apex etc)
  - [ ] Reviewer: Does the PR have correct import guards for all optional libraries?
  
**PR Type**:
- [x] New Feature
- [ ] Bugfix
- [ ] Documentation

If you haven't finished some of the above items you can still open "Draft" PR.


## Who can review?

Anyone in the community is free to review the PR once the checks have passed. 
[Contributor guidelines](https://github.com/NVIDIA/NeMo/blob/main/CONTRIBUTING.md) contains specific people who can review PRs to various areas.

# Additional Information
* Related to # (issue)
