Add Aumann-Shapley sensitivity scoring method to auto_quantize - #2183

Open

joshua-hill wants to merge 13 commits into

NVIDIA:mainfrom

joshua-hill:feat/aumann-shapley-autoquant

Commits on Aug 22, 2026

Refactor AutoQuantize backward scoring
joshua-hill
committed
cb93b2d

Commits on Aug 23, 2026

Restore AutoQuantize forward attributes exactly
joshua-hill
committed
d733339

Commits on Aug 24, 2026

Restore AutoQuantize backward hook state
joshua-hill
committed
2edb3a3

Commits on Aug 25, 2026

Fix gradient scoring for reused modules
joshua-hill
committed
70acb63
Guard gradient replay stacks by autograd participation
joshua-hill
committed
14b2af2
Bind replay scoring to output invocations
joshua-hill
committed
00aba57
Avoid scalar extraction in distributed scoring test
joshua-hill
committed
e34f072
Bound output hooks to autograd graph lifetime
joshua-hill
committed
d5a62bd
Add Aumann-Shapley AutoQuantize scoring
joshua-hill
committed
f12632e
Avoid duplicate AutoQuantize score warning
joshua-hill
committed
f3f7869
Fix Aumann-Shapley damage-bound constraints
joshua-hill
committed
b8cc6ce
Fix Aumann-Shapley scoring edge cases
joshua-hill
committed
f373136
Merge branch 'main' into feat/aumann-shapley-autoquant
joshua-hill
authored
c746da0