[megatron] Accept dtype-string optimizer_config_kwargs (coerce exp_avg_dtype etc. to torch.dtype)#1805

Open

dyurk-lila wants to merge 2 commits into

NovaSky-AI:mainfrom

dyurk-lila:feat/optimizer-state-dtype-coercion

Commits on Jul 1, 2026

[megatron] Accept dtype-string optimizer_config_kwargs (coerce exp_avg_dtype etc. to torch.dtype)
dyurk-lila
committed
docs: tighten optimizer dtype comments
dyurk-lila
committed