accelerate

Use device as context manager for init_on_device

#1826

Merged