lr scheduler - update global states into optimizer

[lessw2020]

lr scheduler currently maintains two global states to implement the full lr warmup and decay.
We want to remove these:
"nit: we can make these two arguments still as function arguments below, but we use:

functools.partial to pack these two arguments when adding to the optimizer.

global config works for now. This can be done as a follow up"

[awgu]

cc: @tianyu-l is this issue done?

[tianyu-l]

Seems not. @awgu

https://github.com/pytorch/torchtitan/blob/main/torchtitan/lr_scheduling.py#L10

cc: @tianyu-l is this issue done?