accelerate

Add kwargs to optimizer, scheduler and dataloader using function `accelerator().load_state()`

#3540

Merged