Implementation of a more flexible optimizer/scheduler override system - #2723

Merged

ko3n1g merged 17 commits into

NVIDIA:mainfrom

jstjohn:optimizer_param_lookup_refactor

Jan 7, 2026

Commits on Dec 19, 2025

Implementation of a more flexible optimizer/scheduler override system
jstjohn
committed
fc2fc3d
Add more comments to get param groups since the behavior is different (no longer 1-1 constraint)
jstjohn
committed
9e81f47
Fix circular import in optimizer_param_scheduler
jstjohn
committed
856e395
Add test coverage
jstjohn
committed
6c816c6
Adding test coverage for ParamKey and ParamPredicte matching
jstjohn
committed
253b9f7
Also add clarifying test around what happens if an attribute is there but False
jstjohn
committed
8c9725d
Remove unused import and reformatting
jstjohn
committed
640ee26
Add/update more comments
jstjohn
committed
dbaafc5
Show that ordering does not impact the merging of the param group keys
jstjohn
committed
d16efe4

Commits on Jan 5, 2026

Add utility function for returning a standard set of config overrides for the user to make use of
jstjohn
committed
0faf0ea
Have default behavior for users of mcore who do not provide any skips
jstjohn
committed
6ae2c41
Adding missing license headers
jstjohn
committed
7d1ecc3
Update test to cover new default
jstjohn
committed
59b9023
Fix issue for fp8 cards that do not support block scaling
jstjohn
committed
8ca2cf4

Commits on Jan 6, 2026