Dgalvez/wip conditional nodes functional
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 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 ?

Add my_torch_cond function, which emulates torch.cond. Also add support for CUDA Graphs stream capture to my_torch_cond.

This is to support easier implementation of RNN-T greedy inference, and other data-dependent control flow heavy algorithms.

**Collection**: Core, ASR

This is a draft for now. No need for the core nemo people to review it.