Pass `lr_scheduler` to `Accelerator.prepare` (#301)
* Work in progress
* Pass scheduler to Accelerator.prapre
* Fix tests
* Apply suggestions from code review
Co-authored-by: Zachary Mueller <muellerzr@gmail.com>
* Style post comments
Co-authored-by: Zachary Mueller <muellerzr@gmail.com>
