Skip to content

[PyTorch] Move FusedAdam/FusedSGD and necessary kernels from Apex to TE#867

Merged
timmoon10 merged 7 commits into
NVIDIA:mainfrom
yaox12:fused_adam
May 30, 2024
Merged

[PyTorch] Move FusedAdam/FusedSGD and necessary kernels from Apex to TE#867
timmoon10 merged 7 commits into
NVIDIA:mainfrom
yaox12:fused_adam

Commits

Commits on May 27, 2024

Commits on May 28, 2024

Commits on May 30, 2024