[INSTALL-9] Transformer Engine uv install should be unified with the main NeMo-RL install
Open
Documentation
p1
- Dominant language
- Python
- Stars
- 2k
- Forks
- 561
- Avg merge
- 4d 5h
- Merged PRs (30d)
- 145
Description
## Problem
The installation guide requires a separate `uv` command to install Transformer Engine with the correct `cudnn` version. It is not clear why this step is separate from the main NeMo-RL install. The split increases friction and the chance of users missing the step.
## Suggested fix
Consolidate TE installation into the main `uv` install flow where possible. If a separate step is unavoidable, add a clear explanation of why and make it visually prominent in the guide.
Contributor guide
Assessment
This issue has not been assessed yet.