Skip to content

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

Merged
SumanthRH merged 4 commits into
NovaSky-AI:mainfrom
dyurk-lila:feat/optimizer-state-dtype-coercion
Jul 10, 2026
Merged

[megatron] Accept dtype-string optimizer_config_kwargs (coerce exp_avg_dtype etc. to torch.dtype)#1805
SumanthRH merged 4 commits into
NovaSky-AI:mainfrom
dyurk-lila:feat/optimizer-state-dtype-coercion

test: make optimizer dtype test CPU-only

6f50dbe
Select commit
Loading
Failed to load commit list.
Sign in for the full log view

Annotations

1 warning
skyrl_tests
succeeded Jul 10, 2026 in 50m 4s