CUDA Graphs

Validate and use CUDA graph capture in Megatron Bridge, including local full-iteration graphs and Transformer Engine scoped graphs for attention, MLP, and MoE modules.

Source: github.com/NVIDIA/skills/tree/HEAD/skills/nemo-mbridge-perf-cuda-graphs · License: Other