Fix/per module accumulate iterations#50

Merged

pomonam merged 8 commits into

mainfrom

fix/per-module-accumulate-iterations

May 23, 2026

Commits on May 23, 2026

Fix accumulate_iterations missing for per-module pairwise scores

pomonam
and
claude
committed
Fix README example using train=True for eval dataset

pomonam
and
claude
committed
Polish README quickstart example

pomonam
and
claude
committed
Fix documentation drift

pomonam
and
claude
committed
Gate score-module GradScaler on score_args.amp_dtype, not factor_args

pomonam
and
claude
committed
Fix off-by-one in target_data_partitions/target_module_partitions bounds check

pomonam
and
claude
committed
Clamp eigenvalues to >=0 after eigh

pomonam
and
claude
committed
Reject compute_per_token_scores=True for Conv2d modules

pomonam
and
claude
committed