[PyTorch] fix cross entropy vanishing gradients - #2139

Merged

timmoon10 merged 12 commits into

NVIDIA:mainfrom

casper-hansen:fix-ce

Sep 4, 2025

Commits on Aug 30, 2025

fix cross entropy
casper-hansen
committed
106c4cc
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot]
authored and
casper-hansen
committed
5ccad3f
fix comments
casper-hansen
committed
988ff30
fix: few more style issues
casper-hansen
committed
8bea946
fix: remove grad_output_stride (unnecessary)
casper-hansen
committed
2f36534
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot]
committed
8269185
fix: only backward was broken
casper-hansen
committed
ac10da4
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot]
committed
c3c7b73

Commits on Sep 3, 2025

Merge branch 'main' into fix-ce
timmoon10
authored
74d5340

Commits on Sep 4, 2025