Fix gradient-norm undercounting when using EP and TP - #5916

Merged

philipcmonk merged 9 commits into

NVIDIA:mainfrom

philipcmonk:expert-norms

Jul 25, 2026

Commits on Jul 21, 2026

fix wgrad/param-norm undercounting when EP and TP are used
philipcmonk
committed
2629ec4
add shared expert to param norm/grad-clipping tests
philipcmonk
committed
ba58aa0

Commits on Jul 23, 2026

clean up _set_expert_parameter_attributes
philipcmonk
committed
f93d347
address review comments
philipcmonk
committed
6d7bc08
remove non-finite value from golden test
philipcmonk
committed
3184ecb

Commits on Jul 24, 2026

fix more golden tests
philipcmonk
committed
40e1603

Commits on Jul 25, 2026

Merge remote-tracking branch 'upstream/main' into expert-norms
philipcmonk
committed
cf61a99
Merge upstream/main into expert-norms
philipcmonk
committed
637e091
Merge upstream/main into expert-norms
philipcmonk
committed
6ccd20e