PR #3081 Ensure that te.pytorch exists

Ensure that te.pytorch exists #3081

githubnemo merged 3 commits into huggingface:main from githubnemo:issue/check-te-pytorch

Ensure that te.pytorch exists

6bfc28d1

githubnemo requested a review from

BenjaminBossan 15 days ago

Add TE to the GPU docker file

00187bd8

BenjaminBossan approved these changes on 2026-03-11

Use CUDNN docker image (TE requirement)

a4bc0a9a

githubnemo merged 04d6a4a2 into main 7 days ago

Reviewers

BenjaminBossan

Assignees

No one assigned

Labels

None yet

Milestone

No milestone