NVIDIA-NeMo / NVIDIA-NeMo/RL

[INSTALL-9] Transformer Engine uv install should be unified with the main NeMo-RL install

Open
#2,713 0 comments 0 reactions 1 assignee Claimed by @kajalj22 View on GitHub
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

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.