nemo-ux cudagraph
This PR is stale because it has been open for 14 days with no activity. Remove stale label or comment or update or this will be closed in 7 days. This PR was closed because it has been inactive for 7 days since being marked as stale.
# What does this PR do ?

Adds support for cudagraph in NeMo2.0.

Example config:

> gpt_config = llm.GPTConfig(
>        num_layers=1,
>        hidden_size=128,
>        ffn_hidden_size=256,
>        num_attention_heads=1,
>        seq_length=seq_length,
>        init_method_std=0.023,
>        hidden_dropout=0.0,
>        attention_dropout=0.0,
>        layernorm_epsilon=1e-5,
>        make_vocab_size_divisible_by=128,
>        normalization='RMSNorm',
>        masked_softmax_fusion=False,
>        # Cudagraph options
>        enable_cuda_graph=True,
>        use_te_rng_tracker=True,
>        gradient_accumulation_fusion=False,  # This should be false with cudagraphs
> )

**Collection**: [Note which collection this PR will affect]

# 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**:
- [ ] 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)
